ws-bom-robot-app 0.0.43__tar.gz → 0.0.45__tar.gz

{ws_bom_robot_app-0.0.43/ws_bom_robot_app.egg-info → ws_bom_robot_app-0.0.45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ws_bom_robot_app
-Version: 0.0.43
+Version: 0.0.45
 Summary: A FastAPI application serving ws bom/robot/llm platform ai.
 Home-page: https://github.com/websolutespa/bom
 Author: Websolute Spa

{ws_bom_robot_app-0.0.43 → ws_bom_robot_app-0.0.45}/setup.py

@@ -4,7 +4,7 @@ _requirements = [line.split('#')[0].strip() for line in open("requirements.txt")
 
 setup(
     name="ws_bom_robot_app",
-    version="0.0.43",
+    version="0.0.45",
     description="A FastAPI application serving ws bom/robot/llm platform ai.",
     long_description=open("README.md", encoding='utf-8').read(),
     long_description_content_type="text/markdown",

{ws_bom_robot_app-0.0.43 → ws_bom_robot_app-0.0.45}/ws_bom_robot_app/llm/agent_description.py

@@ -88,7 +88,7 @@ class AgentDescriptor:
 
     def __init__(self, llm: LlmInterface, prompt: str, mode: str, rules: LlmRules = None):
         self.__prompt = prompt
-        self.__llm = llm,
+        self.__llm = llm
         self.rules= rules
         self.strategy = self.get_strategy(mode)  # Selects the strategy from the dictionary
 

{ws_bom_robot_app-0.0.43 → ws_bom_robot_app-0.0.45}/ws_bom_robot_app/llm/agent_handler.py

@@ -64,7 +64,7 @@ class AgentHandler(AsyncCallbackHandler):
         if token:
             token = _parse_token(self.llm,token)
             if token:
-              self.stream_buffer += token.replace('\n','')  # append new data to pending buffer
+              self.stream_buffer += token  # append new data to pending buffer
               if not self.in_json_block:
                 # search for the start of a json block.
                 start_match = self.json_start_regex.search(self.stream_buffer)
@@ -90,7 +90,7 @@ class AgentHandler(AsyncCallbackHandler):
                         self.stream_buffer = self.stream_buffer[-3:]
               else:
                   # in json block: look for an end fence.
-                  end_match = self.json_end_regex.search(self.stream_buffer,endpos=3)
+                  end_match = self.json_end_regex.search(self.stream_buffer)
                   if end_match:
                       end_index = end_match.start()
                       self.json_buffer += self.stream_buffer[:end_index]

{ws_bom_robot_app-0.0.43 → ws_bom_robot_app-0.0.45}/ws_bom_robot_app/llm/api.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Any
+from typing import Annotated, Any, Mapping
 from fastapi import APIRouter, HTTPException, Request, Header
 from fastapi.responses import StreamingResponse
 from ws_bom_robot_app.llm.agent_description import AgentDescriptor
@@ -9,7 +9,7 @@ from ws_bom_robot_app.llm.vector_store.generator import kb, rules, kb_stream_fil
 from ws_bom_robot_app.llm.tools.tool_manager import ToolManager
 from ws_bom_robot_app.llm.vector_store.integration.manager import IntegrationManager
 from ws_bom_robot_app.task_manager import task_manager, TaskHeader
-
+from uuid import uuid4
 router = APIRouter(prefix="/api/llm", tags=["llm"])
 
 @router.get("/")
@@ -20,13 +20,15 @@ async def root():
 async def _invoke(rq: InvokeRequest):
     return await invoke(rq)
 
+def _stream_headers(rq: StreamRequest) -> Mapping[str, str]:
+    return {"X-thread-id": rq.thread_id or str(uuid4())}
 @router.post("/stream")
 async def _stream(rq: StreamRequest, ctx: Request) -> StreamingResponse:
-    return StreamingResponse(stream(rq, ctx), media_type="application/json")
+    return StreamingResponse(stream(rq, ctx), media_type="application/json", headers=_stream_headers(rq))
 
 @router.post("/stream/raw")
 async def _stream_raw(rq: StreamRequest, ctx: Request) -> StreamingResponse:
-    return StreamingResponse(stream(rq, ctx, formatted=False), media_type="application/json")
+    return StreamingResponse(stream(rq, ctx, formatted=False), media_type="application/json", headers=_stream_headers(rq))
 
 @router.post("/kb")
 async def _kb(rq: KbRequest) -> VectorDbResponse:

{ws_bom_robot_app-0.0.43 → ws_bom_robot_app-0.0.45}/ws_bom_robot_app/llm/providers/llm_manager.py

@@ -71,10 +71,11 @@ class DeepSeek(LlmInterface):
         )
 
     def get_models(self):
-        return [
-              {"id":"deepseek-chat"},
-              {"id":"deepseek-reasoner"}
-            ]
+        import openai
+        openai.api_key = self.config.api_key or os.getenv("DEEPSEEK_API_KEY")
+        openai.base_url = "https://api.deepseek.com"
+        response = openai.models.list()
+        return response.data
 
 class Google(LlmInterface):
   def get_llm(self):

{ws_bom_robot_app-0.0.43 → ws_bom_robot_app-0.0.45/ws_bom_robot_app.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ws_bom_robot_app
-Version: 0.0.43
+Version: 0.0.45
 Summary: A FastAPI application serving ws bom/robot/llm platform ai.
 Home-page: https://github.com/websolutespa/bom
 Author: Websolute Spa