ok

app/api/v1/router.py

@@ -15,4 +15,5 @@ api_v1_router.include_router(code_executor.router, tags=["Code Executor"])
 api_v1_router.include_router(verify_router, prefix="/verify", tags=["Verify"])
 api_v1_router.include_router(reconcile.router, tags=["Reconcile"])
 api_v1_router.include_router(scraper.router, tags=["Web Scraping"])
+api_v1_router.include_router(web_search.router, tags=["Web Search"])
 api_v1_router.include_router(chat.router, tags=["Chat"])

app/services/chat_service.py

@@ -456,6 +456,7 @@ async def chat_completion(
                 max_tokens, temperature, top_p,
             )
         if result:
+            result["model"] = model
             return result
         raise RuntimeError("AionLabs request failed")
 
@@ -464,6 +465,7 @@ async def chat_completion(
             messages, response_format, max_tokens, temperature, top_p,
         )
         if result:
+            result["model"] = model
             return result
         raise RuntimeError("OpenRouter Mimika request failed")
 
@@ -477,6 +479,7 @@ async def chat_completion(
             messages, response_format, max_tokens, temperature, top_p, target,
         )
     if result:
+        result["model"] = model
         return result
 
     logger.info("MegaNova failed, falling back to AionLabs")
@@ -491,6 +494,7 @@ async def chat_completion(
             max_tokens, temperature, top_p,
         )
     if result:
+        result["model"] = model
         return result
 
     logger.info("AionLabs failed, falling back to OpenRouter Mimika")
@@ -498,6 +502,7 @@ async def chat_completion(
         messages, response_format, max_tokens, temperature, top_p,
     )
     if result:
+        result["model"] = model
         return result
 
     raise RuntimeError("All AI providers exhausted")