ws-bom-robot-app 0.0.31__py3-none-any.whl → 0.0.32__py3-none-any.whl

ws_bom_robot_app/llm/main.py

@@ -42,7 +42,7 @@ async def __stream(rq: StreamRequest,queue: Queue,formatted: bool = True) -> Non
           settings.chat_history.append(HumanMessage(content=message.content))
       elif message.role == "assistant":
           message_content = ""
-          if '{\"type\":\"string\"' in message.content:
+          if '{\"type\":\"text\"' in message.content:
             try:
               json_msg = json.loads('[' + message.content[:-1] + ']')
               for msg in json_msg:
@@ -75,13 +75,13 @@ async def __stream(rq: StreamRequest,queue: Queue,formatted: bool = True) -> Non
         )
     callbacks.append(nebuly_callback)
 
-  with warnings.catch_warnings():
-    warnings.simplefilter("ignore", UserWarning)
+  #with warnings.catch_warnings():
+  #  warnings.simplefilter("ignore", UserWarning)
 
-    await processor.executor.ainvoke(
-        {"input": rq.messages[-1], "chat_history": settings.chat_history},
-        {"callbacks": callbacks},
-    )
+  await processor.executor.ainvoke(
+      {"input": rq.messages[-1], "chat_history": settings.chat_history},
+      {"callbacks": callbacks},
+  )
 
   # Signal the end of streaming
   await queue.put(None)

ws_bom_robot_app/llm/models/api.py

@@ -8,7 +8,7 @@ from ws_bom_robot_app.config import Settings, config
 
 class LlmMessage(BaseModel):
   role: str
-  content: str
+  content: Union[str, list]
 
 class LlmSearchSettings(BaseModel):
   search_type: Optional[str] = Field('default', validation_alias=AliasChoices("searchType","search_type"))

ws_bom_robot_app/llm/vector_store/integration/jira.py

@@ -7,6 +7,8 @@ from langchain_core.documents import Document
 from ws_bom_robot_app.llm.vector_store.loader.base import Loader
 from pydantic import BaseModel, Field, AliasChoices
 from typing import Optional, Union
+import requests
+import unstructured_ingest.connector.jira
 
 class JiraParams(BaseModel):
   """
@@ -26,13 +28,18 @@ class JiraParams(BaseModel):
   projects: list[str]
   boards: Optional[list[str]] | None = None
   issues: Optional[list[str]] | None = None
+  fieldsMappingUrl: Optional[str] | None = None
+
 class Jira(IntegrationStrategy):
+  DEFAULT_C_SEP = " " * 5
+  DEFAULT_R_SEP = "\n"
   def __init__(self, knowledgebase_path: str, data: dict[str, Union[str,int,list]]):
     super().__init__(knowledgebase_path, data)
     self.__data = JiraParams.model_validate(self.data)
   def working_subdirectory(self) -> str:
     return 'jira'
   def run(self) -> None:
+    unstructured_ingest.connector.jira._get_dropdown_fields_for_issue = self._get_dropdown_fields_for_issue
     access_config = JiraAccessConfig(
       api_token=self.__data.access_token
     )
@@ -56,3 +63,52 @@ class Jira(IntegrationStrategy):
       await asyncio.to_thread(self.run)
       await asyncio.sleep(1)
       return await Loader(self.working_directory).load()
+
+  def _remap_custom_fields(self, field_list):
+    auth = (self.__data.user_email, self.__data.access_token)
+    response = requests.get(self.__data.fieldsMappingUrl, auth=auth)
+
+    if response.status_code == 200:
+      mapper: dict = response.json()
+    remapped_field_list = {}
+    for field_key, field_value in field_list.items():
+        new_key = None
+        for map_item in mapper:
+            if field_key == map_item["id"]:
+                # Usa il nome mappato come nuova chiave
+                new_key = map_item["name"]
+                break
+
+        if new_key is None:
+            new_key = field_key
+
+        remapped_field_list[new_key] = field_value
+
+    return remapped_field_list
+
+  def _get_dropdown_fields_for_issue(self, issue, c_sep=DEFAULT_C_SEP, r_sep=DEFAULT_R_SEP):
+      all_fields = {}
+      for key, value in issue.items():
+          if value is not None:
+              if isinstance(value, list) and (len(value) > 0):
+                  all_fields[key] = value
+              else:
+                  all_fields[key] = value
+      mapped_fields = self._remap_custom_fields(all_fields)
+      return f"""
+      IssueType:{issue["issuetype"]["name"]}
+      {r_sep}
+      Status:{issue["status"]["name"]}
+      {r_sep}
+      Priority:{issue["priority"]}
+      {r_sep}
+      AssigneeID_Name:{issue["assignee"]["accountId"]}{c_sep}{issue["assignee"]["displayName"]}
+      {r_sep}
+      ReporterAdr_Name:{issue["reporter"]["emailAddress"]}{c_sep}{issue["reporter"]["displayName"]}
+      {r_sep}
+      Labels:{c_sep.join(issue["labels"])}
+      {r_sep}
+      Components:{c_sep.join([component["name"] for component in issue["components"]])}
+      {r_sep}
+      {(r_sep + c_sep ).join([f"{key}:{value}{r_sep}" for key, value in mapped_fields.items()])}
+      """

{ws_bom_robot_app-0.0.31.dist-info → ws_bom_robot_app-0.0.32.dist-info}/METADATA

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

@@ -11,10 +11,10 @@ ws_bom_robot_app/llm/agent_handler.py,sha256=7b-H6PCkeFt4gDFv4oXO_Mg6A59MepWm0qw
 ws_bom_robot_app/llm/agent_lcel.py,sha256=O4FgzvKXuIP9VJgSBQtP26Xifx1r-fydD0LtlqUYzug,2730
 ws_bom_robot_app/llm/api.py,sha256=vBu_TFTlBjp7e3J-WmlZbXn_TbB550x-NpQN4YsO7To,3004
 ws_bom_robot_app/llm/defaut_prompt.py,sha256=CDsM6I6vOTklOKD1FK0v93P4TLPjdq7iCQ7wszCs6yE,765
-ws_bom_robot_app/llm/main.py,sha256=VL1xa8i28g4xpLdsxAOnnm-PlExJo1KoIlrGWUNe1jQ,4045
+ws_bom_robot_app/llm/main.py,sha256=OgjWceJca1d3zd9TJHfdalcXAZMCyr8aO6VZGuqLWsY,4037
 ws_bom_robot_app/llm/settings.py,sha256=DCLaGZwxlw0xE46LpfUgin_FHD8_XJIthCgI6r2UDlM,121
 ws_bom_robot_app/llm/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/models/api.py,sha256=14CPG9hCu2rDMNrUGpvi9SOfKr-ALSBZfKtliSjnwEs,6570
+ws_bom_robot_app/llm/models/api.py,sha256=ebyzuXcNNan8xzR94SfediHMJ5PymY4GoPYlfFQPdq4,6583
 ws_bom_robot_app/llm/models/base.py,sha256=1TqxuTK3rjJEALn7lvgoen_1ba3R2brAgGx6EDTtDZo,152
 ws_bom_robot_app/llm/models/kb.py,sha256=9zqwDlVULVrWE48wo5AivzWoOtnjA57k9rsw8KNnyDk,8935
 ws_bom_robot_app/llm/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -41,7 +41,7 @@ ws_bom_robot_app/llm/vector_store/integration/dropbox.py,sha256=yhGvHTN0TEpUfhdv
 ws_bom_robot_app/llm/vector_store/integration/gcs.py,sha256=fFDVDUR6eNB7FVTzDSEpMHFEWMgG16GLnpSf_mqGDdE,3184
 ws_bom_robot_app/llm/vector_store/integration/github.py,sha256=18PO30AZcgTn6PHhid3MwImVAdmKBNkr0kmAPgOetGw,2663
 ws_bom_robot_app/llm/vector_store/integration/googledrive.py,sha256=R6hr8iEgrR3QMOzIj5jY6w1x8pZ1LGdh4xM_q7g_ttc,3738
-ws_bom_robot_app/llm/vector_store/integration/jira.py,sha256=o5iINIblp_yNszp54nf7fW97aqjs0A5G89N8sYrd1ds,2771
+ws_bom_robot_app/llm/vector_store/integration/jira.py,sha256=uiNkCOSqxOgm3AoO6k5TN_bsYXiO60I_wF0i_z8Yp8Y,4874
 ws_bom_robot_app/llm/vector_store/integration/manager.py,sha256=5Fl3XML6f1wmgraigpUwIFIXh7QFPX0RI0YFgFxBAvg,1700
 ws_bom_robot_app/llm/vector_store/integration/s3.py,sha256=3kh-VmH84IW7DdSLvOk6td1VBJ9aohlVJsk5F3cYj0U,3320
 ws_bom_robot_app/llm/vector_store/integration/sftp.py,sha256=WNzjjS1EUykgFB-8e7QkecSa1r1jTJqKyGzR25uJCtM,2848
@@ -52,7 +52,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=Bv3r5YYAjLHp4sU_sxTk6-OmUdEgoVDqKL-xgWD9k_s,5240
 ws_bom_robot_app/llm/vector_store/loader/docling.py,sha256=12sMSH8DkEsC1Ctml2EIX2gs1BDnWWdynUEqGv-JAF4,2114
 ws_bom_robot_app/llm/vector_store/loader/json_loader.py,sha256=LDppW0ZATo4_1hh-KlsAM3TLawBvwBxva_a7k5Oz1sc,858
-ws_bom_robot_app-0.0.31.dist-info/METADATA,sha256=I8kp-oQ6Kl4hb8VNLlFwwcnKnyzNQrLb907OeoVLlB4,7756
-ws_bom_robot_app-0.0.31.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ws_bom_robot_app-0.0.31.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
-ws_bom_robot_app-0.0.31.dist-info/RECORD,,
+ws_bom_robot_app-0.0.32.dist-info/METADATA,sha256=2k0lbPSs0i08ckNPoaohVty_UBeN7pIjpFfX1bLZ190,7756
+ws_bom_robot_app-0.0.32.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ws_bom_robot_app-0.0.32.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
+ws_bom_robot_app-0.0.32.dist-info/RECORD,,