Update core/agents/model_doc_sync.py

core/agents/model_doc_sync.py

@@ -150,32 +150,35 @@ def store_pricing(model: str, data: Dict[str, Any]):
         return False
 
 def sync_pricing(model_names: list):
-    """Added verification step"""
+    """Fetch, summarize, and store model pricing info"""
+    from typing import Dict, Any
+    import logging
+
+    logging.basicConfig(level=logging.INFO)
+    index = init_pinecone()
+
     for model in model_names:
         try:
             model = model.strip()
             logging.info(f"\n🔄 Syncing {model}...")
-            
+
             docs = search_model_docs(model)
             if not docs:
                 logging.warning(f"No docs found for {model}")
                 continue
-                
+
             pricing = summarize_pricing(docs)
             if not pricing:
+                logging.error(f"No pricing extracted for {model}")
                 continue
-                
-            # Store and verify
-            if store_pricing(model, pricing):
-                # Immediate fetch verification
-                index = init_pinecone()  # Add this line inside the loop
-                stored_data = index.fetch(ids=[model.lower()])
 
+            if store_pricing(model, pricing):
+                stored_data = index.fetch(ids=[model.lower()], namespace="pricing")
                 if stored_data.vectors:
                     logging.info(f"✅ Verified storage for {model}")
                 else:
                     logging.error(f"❌ Storage failed for {model}")
-            
+
         except Exception as e:
             logging.error(f"Critical error with {model}: {str(e)}")