ws-bom-robot-app 0.0.62__py3-none-any.whl → 0.0.64__py3-none-any.whl

ws_bom_robot_app/config.py

@@ -27,6 +27,7 @@ class Settings(BaseSettings):
     OLLAMA_API_URL: str = 'http://localhost:11434'
     GROQ_API_KEY: str = ''
     GOOGLE_API_KEY: str = ''
+    WATSONX_APIKEY: str = '' # used for ibm watsonx
     NEBULY_API_URL: str =''
     GOOGLE_APPLICATION_CREDENTIALS: str = '' # path to google credentials iam file, e.d. ./.secrets/google-credentials.json
     model_config = ConfigDict(
@@ -44,6 +45,7 @@ class Settings(BaseSettings):
         os.environ["GROQ_API_KEY"] = self.GROQ_API_KEY
         os.environ["GOOGLE_API_KEY"] = self.GOOGLE_API_KEY
         os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = self.GOOGLE_APPLICATION_CREDENTIALS
+        os.environ["WATSONX_APIKEY"] = self.WATSONX_APIKEY
         os.environ["NEBULY_API_URL"] = self.NEBULY_API_URL
 
     class RuntimeOptions(BaseModel):

ws_bom_robot_app/llm/main.py

@@ -89,6 +89,8 @@ async def __stream(rq: StreamRequest, ctx: Request, queue: Queue,formatted: bool
     callbacks.append(trace)
 
   __llm: LlmInterface =rq.get_llm()
+  for tool in rq.app_tools:
+    tool.thread_id = rq.thread_id
   processor = AgentLcel(
       llm=__llm,
       sys_message=rq.system_message,

ws_bom_robot_app/llm/models/api.py

@@ -36,6 +36,7 @@ class LlmAppToolDbSettings(BaseModel):
 
 class LlmAppTool(BaseModel):
   id: Optional[str] = None
+  thread_id: Optional[str] = Field(None, validation_alias=AliasChoices("threadId","thread_id"))
   name: str
   description: Optional[str] = None
   type: str

ws_bom_robot_app/llm/providers/llm_manager.py

@@ -37,6 +37,31 @@ class LlmInterface:
         from langchain.agents.output_parsers.openai_tools import OpenAIToolsAgentOutputParser
         return OpenAIToolsAgentOutputParser()
 
+class Anthropic(LlmInterface):
+    def get_llm(self):
+        from langchain_anthropic import ChatAnthropic
+        return ChatAnthropic(
+            api_key=self.config.api_key or os.getenv("ANTHROPIC_API_KEY"),
+            model=self.config.model,
+            temperature=self.config.temperature,
+            streaming=True,
+            stream_usage=True
+        )
+
+    """
+    def get_embeddings(self):
+        from langchain_voyageai import VoyageAIEmbeddings
+        return VoyageAIEmbeddings(
+            api_key=self.config.embedding_api_key, #voyage api key
+            model="voyage-3")
+    """
+
+    def get_models(self):
+        import anthropic
+        client = anthropic.Client(api_key=self.config.api_key or os.getenv("ANTHROPIC_API_KEY"))
+        response = client.models.list()
+        return response.data
+
 class OpenAI(LlmInterface):
     def __init__(self, config: LlmConfig):
         super().__init__(config)
@@ -133,31 +158,6 @@ class Gvertex(LlmInterface):
               {"id":"gemini-1.5-pro-002"}
             ]
 
-class Anthropic(LlmInterface):
-    def get_llm(self):
-        from langchain_anthropic import ChatAnthropic
-        return ChatAnthropic(
-            api_key=self.config.api_key or os.getenv("ANTHROPIC_API_KEY"),
-            model=self.config.model,
-            temperature=self.config.temperature,
-            streaming=True,
-            stream_usage=True
-        )
-
-    """
-    def get_embeddings(self):
-        from langchain_voyageai import VoyageAIEmbeddings
-        return VoyageAIEmbeddings(
-            api_key=self.config.embedding_api_key, #voyage api key
-            model="voyage-3")
-    """
-
-    def get_models(self):
-        import anthropic
-        client = anthropic.Client(api_key=self.config.api_key or os.getenv("ANTHROPIC_API_KEY"))
-        response = client.models.list()
-        return response.data
-
 class Groq(LlmInterface):
     def get_llm(self):
         from langchain_groq import ChatGroq
@@ -179,10 +179,63 @@ class Groq(LlmInterface):
         response = requests.get(url, headers=headers)
         return response.json().get("data", [])
 
+class IBM(LlmInterface):
+    def __init__(self, config: LlmConfig):
+        super().__init__(config)
+        self.__apy_key = self.config.api_key or os.getenv("WATSONX_APIKEY")
+        self.__base_url = self.config.api_url or "https://us-south.ml.cloud.ibm.com"
+    def get_llm(self):
+        from langchain_ibm import ChatWatsonx
+        return ChatWatsonx(
+            model_id=self.config.model,
+            url=self.__base_url,
+            apikey=self.__apy_key
+        )
+    def get_models(self):
+        import requests
+        from datetime import date
+        try:
+          # https://cloud.ibm.com/apidocs/watsonx-ai#list-foundation-model-specs
+          today = date.today().strftime("%Y-%m-%d")
+          url = f"{self.__base_url}/ml/v1/foundation_model_specs?version={today}&filters=task_generation,task_summarization:and"
+          headers = {
+              "Authorization": f"Bearer {self.__apy_key}",
+              "Content-Type": "application/json"
+          }
+          response = requests.get(url, headers=headers)
+          models = response.json().get("resources", [])
+          return [{
+                "id": model['model_id'],
+                "provider": model['provider'],
+                "tasks": model['task_ids'],
+                "limits": model.get('model_limits', {}),
+              } for model in models if model['provider'].lower() in ['ibm','meta','mistral ai']]
+        except Exception as e:
+          print(f"Error fetching models from IBM WatsonX: {e}")
+          # https://www.ibm.com/products/watsonx-ai/foundation-models
+          return [
+            {"id":"granite-3-3-8b-instruct"},
+            {"id":"granite-vision-3-2-2b"},
+            {"id":"granite-13b-instruct"},
+            {"id":"llama-4-scout-17b-16e-instruct"},
+            {"id":"llama-3-3-70b-instruct"},
+            {"id":"mistral-medium-2505"},
+            {"id":"mistral-small-3-1-24b-instruct-2503"},
+            {"id":"mistral-large-2"}
+          ]
+
+    def get_embeddings(self):
+        from langchain_ibm import WatsonxEmbeddings
+        return WatsonxEmbeddings(
+            model_id="ibm/granite-embedding-107m-multilingual", #https://www.ibm.com/products/watsonx-ai/foundation-models
+            url=self.__base_url,
+            apikey=self.__apy_key
+        )
+
 class Ollama(LlmInterface):
     def __init__(self, config: LlmConfig):
         super().__init__(config)
-        self.__base_url = self.config.api_url or os.getenv("OLLAMA_API_URL")
+        self.__base_url = self.config.api_url or os.getenv("OLLAMA_API_URL") or "http://localhost:11434"
     def get_llm(self):
         from langchain_ollama.chat_models import ChatOllama
         return ChatOllama(
@@ -221,6 +274,7 @@ class LlmManager:
         "google": Google,
         "gvertex": Gvertex,
         "groq": Groq,
+        "ibm": IBM,
         "openai": OpenAI,
         "ollama": Ollama
     }

ws_bom_robot_app/llm/tools/tool_manager.py

@@ -228,7 +228,7 @@ class ToolManager:
         try:
           raw_results = search.results(query, max_results=10)
         except Exception as e:
-            print(f"[!] Errore ricerca: {e}")
+            return f"[!] Errore ricerca: {e}"
         urls = [r["link"] for r in raw_results]
         return await self._fetch_urls(urls)
 
@@ -260,6 +260,11 @@ class ToolManager:
         from email.mime.text import MIMEText
         secrets = self.app_tool.secrets
         secrets = {item["secretId"]: item["secretValue"] for item in secrets}
+        import urllib.parse as urlparse
+        url_preview = secrets.get("url_preview", "")
+        if url_preview and url_preview != "":
+          message_tread = "Puoi visualizzare la chat su questo indirizzo: " + urlparse.urljoin(url_preview, f"?llmThreadId={self.app_tool.thread_id}")
+          body = body.replace("##url_preview##", message_tread)
         # Email configuration
         smtp_server = secrets.get("smtp_server")
         smtp_port = secrets.get("smtp_port")

ws_bom_robot_app/main.py

@@ -70,7 +70,7 @@ def __get_disk_info():
     return _disks
 @app.get("/api/diag",tags=["diag"])
 def diag(authenticate: bool = Depends(authenticate)):
-    import pkg_resources, psutil
+    import importlib,psutil
     from ws_bom_robot_app.llm.providers.llm_manager import LlmManager as wsllm
     from ws_bom_robot_app.llm.vector_store.db.manager import VectorDbManager as wsdb
     from ws_bom_robot_app.llm.vector_store.loader.base import Loader as wsldr
@@ -81,7 +81,7 @@ def diag(authenticate: bool = Depends(authenticate)):
     svmem = psutil.virtual_memory()
     swap = psutil.swap_memory()
     try:
-      ws_bom_robot_app_version = pkg_resources.get_distribution("ws_bom_robot_app").version
+      ws_bom_robot_app_version = importlib.metadata.version("ws_bom_robot_app")
     except:
       ws_bom_robot_app_version = "unknown"
     peer_process_ids = [c.pid for c in psutil.Process(os.getppid()).children()] if config.runtime_options().is_multi_process else None

{ws_bom_robot_app-0.0.62.dist-info → ws_bom_robot_app-0.0.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ws_bom_robot_app
-Version: 0.0.62
+Version: 0.0.64
 Summary: A FastAPI application serving ws bom/robot/llm platform ai.
 Home-page: https://github.com/websolutespa/bom
 Author: Websolute Spa
@@ -22,6 +22,7 @@ Requires-Dist: langchain-community==0.3.26
 Requires-Dist: langchain-core==0.3.67
 Requires-Dist: langchain-openai==0.3.27
 Requires-Dist: langchain-anthropic==0.3.6
+Requires-Dist: langchain-ibm==0.3.14
 Requires-Dist: langchain-google-genai==2.0.7
 Requires-Dist: langchain-google-vertexai==2.0.27
 Requires-Dist: langchain-groq==0.3.5

{ws_bom_robot_app-0.0.62.dist-info → ws_bom_robot_app-0.0.64.dist-info}/RECORD

@@ -1,8 +1,8 @@
 ws_bom_robot_app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/auth.py,sha256=84nIbmJsMrNs0sxIQGEHbjsjc2P6ZrZZGSn8dkiL6is,895
-ws_bom_robot_app/config.py,sha256=JFO4HrFnzUjFfv5MH0Tzwda0krllEqmTufHZ6J-BgEI,4147
+ws_bom_robot_app/config.py,sha256=izfyxYJE4GHNFNsi3UWDluyJySDwoVx1QMQhMzQIQCw,4260
 ws_bom_robot_app/cron_manager.py,sha256=pFHV7SZtp6GRmmLD9K1Mb1TE9Ev9n5mIiFScrc7tpCo,9221
-ws_bom_robot_app/main.py,sha256=zO3B-v-v9ESASvw8IaQj9Y9hNvNmOxohFmA0R82EybQ,6518
+ws_bom_robot_app/main.py,sha256=1vx0k2fEcE53IC5zcE2EUCwQPcUHM4pvuKSun_E0a9I,6501
 ws_bom_robot_app/task_manager.py,sha256=Q3Il2TtkP0FoG9zHEBu48pZGXzimTtvWQsoH6wdvQs0,16077
 ws_bom_robot_app/util.py,sha256=RjVD6B9sHje788Lndqq5DHy6TJM0KLs9qx3JYt81Wyk,4834
 ws_bom_robot_app/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -12,21 +12,21 @@ ws_bom_robot_app/llm/agent_handler.py,sha256=BQ-f--Z5QCJDp-7tzSG_CKrANUCqG65S09p
 ws_bom_robot_app/llm/agent_lcel.py,sha256=GGZcGBKsSBbZQ-_MPI3NUMvT7lTerYgwKs3o74stwSU,2252
 ws_bom_robot_app/llm/api.py,sha256=2bF-UFczY9LuBqPxKObM0TOWYbZgVztX1RiIz5MSorU,5042
 ws_bom_robot_app/llm/defaut_prompt.py,sha256=LlCd_nSMkMmHESfiiiQYfnJyB6Pp-LSs4CEKdYW4vFk,1106
-ws_bom_robot_app/llm/main.py,sha256=-7HVVpVa_SIDrZS1zS92moQJ0SOSaBrMIPKOdhQ1jEQ,5006
+ws_bom_robot_app/llm/main.py,sha256=NlrEi3fRuNAyHeyOXSZvjSWePSLekahNOZoJ2XrbBbA,5068
 ws_bom_robot_app/llm/nebuly_handler.py,sha256=WDKDx7ItBv39dhAkYtRciA11YWUwZ7HjEOI2cgr-5NI,7851
 ws_bom_robot_app/llm/settings.py,sha256=EkFGCppORenStH9W4e6_dYvQ-5p6xiEMpmUHBqNqG9M,117
 ws_bom_robot_app/llm/feedbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/feedbacks/feedback_manager.py,sha256=WcKgzlOb8VFG7yqHoIOO_R6LAzdzE4YIRFCVOGBSgfM,2856
 ws_bom_robot_app/llm/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/models/api.py,sha256=9GK35esx-jl4EIkpSGbiAOyYPbdhtB59txwoyAgdwnk,10939
+ws_bom_robot_app/llm/models/api.py,sha256=rsPH1y-hMkwXnvt5344yIDMY_StbbgB5AS3Gz37bOhs,11036
 ws_bom_robot_app/llm/models/base.py,sha256=1TqxuTK3rjJEALn7lvgoen_1ba3R2brAgGx6EDTtDZo,152
 ws_bom_robot_app/llm/models/feedback.py,sha256=zh1jLqPRLzNlxInkCMoiJbfSu0-tiOEYHM7FhC46PkM,1692
 ws_bom_robot_app/llm/models/kb.py,sha256=oVSw6_dmNxikAHrPqcfxDXz9M0ezLIYuxpgvzfs_Now,9514
 ws_bom_robot_app/llm/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/providers/llm_manager.py,sha256=zIkxgTLYQCcup2Ixf4eWap4mNinuJH2YmkjLjZGDyJM,8371
+ws_bom_robot_app/llm/providers/llm_manager.py,sha256=8lScD-tG1o2g_tM04Ju96j2JyLIAa5Z8Gm5ttQfDOFQ,10750
 ws_bom_robot_app/llm/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/tools/tool_builder.py,sha256=p5CeLCuAilhtEAbPaiAWKGtuIWl8vfilZjYJ9Kw1dLg,3200
-ws_bom_robot_app/llm/tools/tool_manager.py,sha256=6P8YinxI3RYcwkyFTqZkX9C3Lu43Kl0-o6pwQRj3pPo,13719
+ws_bom_robot_app/llm/tools/tool_manager.py,sha256=Z7yJSIn86BvGtWCkpT48pbu59qLpNe6XtpnoFI-jQU8,14073
 ws_bom_robot_app/llm/tools/utils.py,sha256=tdmOAk8l4HVzw67z3brA9yX-1WLu91paU-WmXHyz4Bg,1883
 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
@@ -66,7 +66,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.62.dist-info/METADATA,sha256=b1mmWDRfVWVw1PbJmb75XL68KuGe6VnA5Yv-reMGB-4,8459
-ws_bom_robot_app-0.0.62.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ws_bom_robot_app-0.0.62.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
-ws_bom_robot_app-0.0.62.dist-info/RECORD,,
+ws_bom_robot_app-0.0.64.dist-info/METADATA,sha256=vsebKjT-mL9G6m1ob_MrGcbOXJkmbjzzA3RZNNjO-UE,8497
+ws_bom_robot_app-0.0.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ws_bom_robot_app-0.0.64.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
+ws_bom_robot_app-0.0.64.dist-info/RECORD,,