ws-bom-robot-app 0.0.36__py3-none-any.whl → 0.0.37__py3-none-any.whl

ws_bom_robot_app/llm/main.py

@@ -1,22 +1,20 @@
-from typing import  AsyncGenerator
-
+from asyncio import Queue
+import asyncio, json, logging, os, traceback
 from fastapi import Request
-from ws_bom_robot_app.llm.agent_lcel import AgentLcel
-from ws_bom_robot_app.llm.agent_handler import AgentHandler, RawAgentHandler
+from langchain.callbacks.tracers import LangChainTracer
+from langchain_core.callbacks.base import AsyncCallbackHandler
+from langchain_core.messages import AIMessage, HumanMessage
+from langsmith import Client as LangSmithClient
+from nebuly.providers.langchain import LangChainTrackingHandler
+from typing import AsyncGenerator, List
+from ws_bom_robot_app.config import config
 from ws_bom_robot_app.llm.agent_description import AgentDescriptor
-from langchain_core.messages import HumanMessage, AIMessage
-from ws_bom_robot_app.llm.providers.llm_manager import LlmConfig, LlmInterface, LlmManager
-from ws_bom_robot_app.llm.tools.tool_builder import get_structured_tools
+from ws_bom_robot_app.llm.agent_handler import AgentHandler, RawAgentHandler
+from ws_bom_robot_app.llm.agent_lcel import AgentLcel
 from ws_bom_robot_app.llm.models.api import InvokeRequest, StreamRequest
+from ws_bom_robot_app.llm.providers.llm_manager import LlmInterface
+from ws_bom_robot_app.llm.tools.tool_builder import get_structured_tools
 import ws_bom_robot_app.llm.settings as settings
-from nebuly.providers.langchain import LangChainTrackingHandler
-from langchain_core.callbacks.base import AsyncCallbackHandler
-import warnings, asyncio, os, io, sys, json,  logging, traceback
-from typing import List
-from asyncio import Queue
-from langchain.callbacks.tracers import LangChainTracer
-from langsmith import Client as LangSmithClient
-from starlette.datastructures import Headers
 
 async def invoke(rq: InvokeRequest) -> str:
   await rq.initialize()
@@ -91,6 +89,9 @@ async def __stream(rq: StreamRequest, ctx: Request, queue: Queue,formatted: bool
   except Exception as e:
     _error = f"Agent invoke ex: {e}"
     logging.warning(_error)
+    if config.runtime_options().debug:
+      _error += f" | {traceback.format_exc()}"
+      await queue.put(_error)
     await queue.put(None)
 
   # Signal the end of streaming

{ws_bom_robot_app-0.0.36.dist-info → ws_bom_robot_app-0.0.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ws_bom_robot_app
-Version: 0.0.36
+Version: 0.0.37
 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.36.dist-info → ws_bom_robot_app-0.0.37.dist-info}/RECORD

@@ -11,7 +11,7 @@ ws_bom_robot_app/llm/agent_handler.py,sha256=4zdpSf5iVLxMZ90c_vUl_k-O9SF6u_h7GOB
 ws_bom_robot_app/llm/agent_lcel.py,sha256=BUfGVUcw6s_YAu5aPMkqqjsStYNHUXKh31t_Ybx11-A,2395
 ws_bom_robot_app/llm/api.py,sha256=UaD1oJyAOe7ASoXxPNJcth3kDuWcjk1xqUNEjuPWbR4,3759
 ws_bom_robot_app/llm/defaut_prompt.py,sha256=LlCd_nSMkMmHESfiiiQYfnJyB6Pp-LSs4CEKdYW4vFk,1106
-ws_bom_robot_app/llm/main.py,sha256=Qw4SbZtstXo31bWFLRQmm_t8BEKlqvgXmtYbxM2F6_Y,4138
+ws_bom_robot_app/llm/main.py,sha256=fVXyS9TOu22ZC7M8o2mRCya9vTmMFf5jRgs9V0K_4cw,4189
 ws_bom_robot_app/llm/settings.py,sha256=EkFGCppORenStH9W4e6_dYvQ-5p6xiEMpmUHBqNqG9M,117
 ws_bom_robot_app/llm/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/models/api.py,sha256=mLbPG7jHh1EjgQG-xpBhEgiTIHpK35HZ51obgqQSfq4,8890
@@ -54,7 +54,7 @@ ws_bom_robot_app/llm/vector_store/loader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 ws_bom_robot_app/llm/vector_store/loader/base.py,sha256=L_ugekNuAq0N9O-24wtlHSNHkqSeD-KsJrfGt_FX9Oc,5340
 ws_bom_robot_app/llm/vector_store/loader/docling.py,sha256=yP0zgXLeFAlByaYuj-6cYariuknckrFds0dxdRcnVz8,3456
 ws_bom_robot_app/llm/vector_store/loader/json_loader.py,sha256=qo9ejRZyKv_k6jnGgXnu1W5uqsMMtgqK_uvPpZQ0p74,833
-ws_bom_robot_app-0.0.36.dist-info/METADATA,sha256=ROer_6Ot24hgee3HuoWzfzQPRzr1JX1XZArffWzZAW0,8348
-ws_bom_robot_app-0.0.36.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ws_bom_robot_app-0.0.36.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
-ws_bom_robot_app-0.0.36.dist-info/RECORD,,
+ws_bom_robot_app-0.0.37.dist-info/METADATA,sha256=UHDhg12pT_4h3GtmCUUzI6zj-Hjso0ATpLRHIC5iOJ4,8348
+ws_bom_robot_app-0.0.37.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ws_bom_robot_app-0.0.37.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
+ws_bom_robot_app-0.0.37.dist-info/RECORD,,