ws-bom-robot-app 0.0.101__py3-none-any.whl → 0.0.102__py3-none-any.whl

ws_bom_robot_app/llm/agent_lcel.py

@@ -22,13 +22,9 @@ class AgentLcel:
         self.embeddings = llm.get_embeddings()
         self.memory_key: str = "chat_history"
         self.__llm_with_tools = llm.get_llm().bind_tools(self.__tools) if len(self.__tools) > 0 else llm.get_llm()
-
-        # Configura l'LLM in base alla modalità
         if self.json_schema:
-            # JSON mode: usa with_structured_output per output strutturato mantenendo i tool
             self.__pydantic_schema = self.__create_pydantic_schema()
         else:
-            # Agent mode standard
             self.__pydantic_schema = None
 
         self.executor = self.__create_agent()
@@ -68,13 +64,7 @@ class AgentLcel:
         return create_model('JsonSchema', **fields)
 
     def __get_output_parser(self):
-        """Restituisce il parser appropriato in base alla modalità."""
-        if self.json_schema and self.__pydantic_schema:
-            # In modalità JSON, usa with_structured_output direttamente nell'agent
-            # Il parser qui serve solo per compatibilità, la conversione avviene prima
-            return self.__llm.get_parser()
-        else:
-            return self.__llm.get_parser()
+        return self.__llm.get_parser()
 
     async def __create_prompt(self, input: dict) -> ChatPromptTemplate:
         from langchain_core.messages import SystemMessage
@@ -82,9 +72,14 @@ class AgentLcel:
         rules_prompt = await get_rules(self.embeddings, self.rules, message.content) if self.rules else ""
         system = default_prompt + (tool_prompt(render_text_description(self.__tools)) if len(self.__tools)>0 else "") + self.sys_message + rules_prompt
 
-        # Aggiungere istruzioni per output strutturato se è presente json_schema
+        # Aggiungi istruzioni per output JSON strutturato se necessario
         if self.json_schema and self.__pydantic_schema:
-            system += f"\n\nYou must respond with a JSON object matching this schema: {self.__pydantic_schema.schema_json()}"
+            json_instructions = f"\n\nIMPORTANT: You must format your final response as a JSON object with the following structure:\n"
+            for field_name, field_info in self.__pydantic_schema.model_fields.items():
+                field_type = field_info.annotation.__name__ if hasattr(field_info.annotation, '__name__') else str(field_info.annotation)
+                json_instructions += f"- {field_name}: {field_type}\n"
+            json_instructions += "\nProvide ONLY the JSON object in your response, no additional text."
+            system += json_instructions
 
         messages = [
             SystemMessage(content=system),
@@ -98,14 +93,15 @@ class AgentLcel:
         )
         return prompt
 
-    def __create_agent(self) -> AgentExecutor:
-      agent: Any = (
-          {
-            "agent_scratchpad": lambda x: self.__llm.get_formatter(x["intermediate_steps"]),
-             str(self.memory_key): lambda x: x[self.memory_key],
-          }
-          | RunnableLambda(self.__create_prompt)
-          | self.__llm_with_tools
-          | self.__get_output_parser()
-      )
-      return AgentExecutor(agent=agent,tools=self.__tools,verbose=False)
+    def __create_agent(self):
+        # Un solo AgentExecutor per entrambe le modalità
+        agent = (
+            {
+                "agent_scratchpad": lambda x: self.__llm.get_formatter(x["intermediate_steps"]),
+                self.memory_key: lambda x: x[self.memory_key],
+            }
+            | RunnableLambda(self.__create_prompt)
+            | self.__llm_with_tools
+            | self.__get_output_parser()
+        )
+        return AgentExecutor(agent=agent, tools=self.__tools, verbose=False)

ws_bom_robot_app/llm/tools/tool_manager.py

@@ -1,5 +1,5 @@
 from asyncio import Queue
-import aiohttp
+import aiohttp, re
 from typing import  Optional, Type, Callable
 from ws_bom_robot_app.config import config
 from ws_bom_robot_app.llm.models.api import LlmApp,LlmAppTool
@@ -200,10 +200,21 @@ class ToolManager:
         else:  # default
           try:
             from ws_bom_robot_app.llm.main import stream
+            import json
             chunks = []
             async for chunk in stream(rq=app.rq, ctx=None, formatted=False):
                 chunks.append(chunk)
             rs = ''.join(chunks) if chunks else None
+
+            # if the app has output_structure, parse the JSON and return dict
+            if rs and app.rq.output_structure:
+                try:
+                    cleaned_rs = re.sub(r'^```(?:json)?\s*\n?', '', rs.strip())
+                    cleaned_rs = re.sub(r'\n?```\s*$', '', cleaned_rs)
+                    return json.loads(cleaned_rs)
+                except json.JSONDecodeError:
+                    print(f"[!] Failed to parse JSON output from proxy_app_chat: {rs}")
+                    return rs
             return rs
           except Exception as e:
             print(f"[!] Error in proxy_app_chat: {e}")

ws_bom_robot_app/llm/utils/cms.py

@@ -37,6 +37,14 @@ async def get_apps() -> list[CmsApp]:
           if obj is None:
               break
       return obj
+  def __to_dict(obj):
+      """Converts DictObject to dict recursively"""
+      if isinstance(obj, DictObject):
+          return {k: __to_dict(v) for k, v in obj.__dict__.items()}
+      elif isinstance(obj, list):
+          return [__to_dict(item) for item in obj]
+      else:
+          return obj
   host = config.robot_cms_host
   if host:
     url = f"{host}/api/llmApp?depth=1&pagination=false&locale=it"
@@ -77,7 +85,8 @@ async def get_apps() -> list[CmsApp]:
                         ) if __attr(_cms_app_dict.settings,'rules','vectorDbFile','filename') else None,
                       #fine_tuned_model=__attr(_cms_app_dict.settings,'llmConfig','fineTunedModel'),
                       lang_chain_tracing= __attr(_cms_app_dict.settings,'llmConfig','langChainTracing', default=False),
-                      lang_chain_project= __attr(_cms_app_dict.settings,'llmConfig','langChainProject', default='')
+                      lang_chain_project= __attr(_cms_app_dict.settings,'llmConfig','langChainProject', default=''),
+                      output_structure= __to_dict(__attr(_cms_app_dict.settings,'llmConfig','outputStructure')) if __attr(_cms_app_dict.settings,'llmConfig','outputStructure') else None
                     ))
                 except Exception as e:
                   import traceback

{ws_bom_robot_app-0.0.101.dist-info → ws_bom_robot_app-0.0.102.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ws_bom_robot_app
-Version: 0.0.101
+Version: 0.0.102
 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.101.dist-info → ws_bom_robot_app-0.0.102.dist-info}/RECORD

@@ -10,7 +10,7 @@ ws_bom_robot_app/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 ws_bom_robot_app/llm/agent_context.py,sha256=uatHJ8wcRly6h0S762BgfzDMpmcwCHwNzwo37aWjeE0,1305
 ws_bom_robot_app/llm/agent_description.py,sha256=5IP0qFSJvaE3zjGS7f0W1DuiegP0RHXRMBoDC5pCofA,4779
 ws_bom_robot_app/llm/agent_handler.py,sha256=HAg3qmj-QY_k7P-GfAKna1tKdmZaTHrZbNWJc0eol3A,7858
-ws_bom_robot_app/llm/agent_lcel.py,sha256=4eOEXCgPX3p6IE9CmfaAAsPJNeGvWA_xF9OZV03Kxo0,4741
+ws_bom_robot_app/llm/agent_lcel.py,sha256=Bkz3JAAR_NibV27H9zfHlleklps93J0mQL53enFoqRo,4694
 ws_bom_robot_app/llm/api.py,sha256=jMoiKiD5HNxGu6gTb5_qZ5UU8d2uJ7UVrdLseDStI6o,7634
 ws_bom_robot_app/llm/defaut_prompt.py,sha256=D9dn8yPveu0bVwGM1wQWLYftmBs5O76o0R_caLLll8w,1121
 ws_bom_robot_app/llm/evaluator.py,sha256=tUyPX1oGZEjSiO4JixwNlgv6BI9cUHSmcAsTCpBnIn4,13322
@@ -27,7 +27,7 @@ ws_bom_robot_app/llm/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 ws_bom_robot_app/llm/providers/llm_manager.py,sha256=oVeEmZUnR1ysV-BI_zpwQ-gpXqmhSzjKFQQAHtaFGSI,16596
 ws_bom_robot_app/llm/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/tools/tool_builder.py,sha256=CtZwJ94aj0YGA3yVWkyCUxNE7WgU2zWjhl_tEfEskxw,3432
-ws_bom_robot_app/llm/tools/tool_manager.py,sha256=avoFERE0v9MFQ3pUBMug8eGYIXbIYl7NqkP1kjNee7s,15439
+ws_bom_robot_app/llm/tools/tool_manager.py,sha256=ThiaBCDOn192NOaLHsxbNZXYs5fptqlfgCHW-9h2eVY,15989
 ws_bom_robot_app/llm/tools/utils.py,sha256=Ba7ScFZPVJ3ke8KLO8ik1wyR2f_zC99Bikqx0OGnKoI,1924
 ws_bom_robot_app/llm/tools/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/tools/models/main.py,sha256=1hICqHs-KS2heenkH7b2eH0N2GrPaaNGBrn64cl_A40,827
@@ -35,7 +35,7 @@ ws_bom_robot_app/llm/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 ws_bom_robot_app/llm/utils/agent.py,sha256=uFuSfYMfGIE2WCKGNSKL-T2SDFn-tUKvbAYbGTPIw6g,1445
 ws_bom_robot_app/llm/utils/chunker.py,sha256=zVXjRMloc3KbNEqiDcycYzy4N0Ey1g8XYeq6ftyvkyg,857
 ws_bom_robot_app/llm/utils/cleanup.py,sha256=ARLZTX4mLbkLCEnMdIWYDYEAPOjzfy1laLGkYnxZe30,3063
-ws_bom_robot_app/llm/utils/cms.py,sha256=5TBDDlTsE4O8_bGvlqFOkkK13WFEoOvYRp_FOEXUuKY,6466
+ws_bom_robot_app/llm/utils/cms.py,sha256=gfIXvY3DxgbgDf0LCzyekWitaduxKGLHfV6gbRmh8zk,6960
 ws_bom_robot_app/llm/utils/download.py,sha256=rvc88E63UGHnFVlJJeMb05Z2FcBYIITqKnIE3ldEu6I,7293
 ws_bom_robot_app/llm/utils/print.py,sha256=HK3zhZOd4cEyXZ8QcudLtTIfqqtMOERce_yTofS8NXo,803
 ws_bom_robot_app/llm/utils/secrets.py,sha256=-HtqLIDVIJrpvGC5YhPAVyLsq8P4ChVM5g3GOfdwqVk,878
@@ -70,7 +70,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=InpRwKPxp0tuM4drezBvxxAWHe3XTmu60MGvFsT7RPE,7176
 ws_bom_robot_app/llm/vector_store/loader/docling.py,sha256=dLvpOi4EH0jyx06IrHoanfLRPFXLJi9BU2BWYcaw-U4,5000
 ws_bom_robot_app/llm/vector_store/loader/json_loader.py,sha256=LDppW0ZATo4_1hh-KlsAM3TLawBvwBxva_a7k5Oz1sc,858
-ws_bom_robot_app-0.0.101.dist-info/METADATA,sha256=FlworyfxD6hsx_BlbG3OC2piu5HXFj5rwSQC3dosCtk,11025
-ws_bom_robot_app-0.0.101.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ws_bom_robot_app-0.0.101.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
-ws_bom_robot_app-0.0.101.dist-info/RECORD,,
+ws_bom_robot_app-0.0.102.dist-info/METADATA,sha256=MOViAkxOwrXz2ynceG8p6KjRYdBI3XPzId3CdZ4SgaI,11025
+ws_bom_robot_app-0.0.102.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ws_bom_robot_app-0.0.102.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
+ws_bom_robot_app-0.0.102.dist-info/RECORD,,