maisaedu-utilities-prefect 1.2.55__tar.gz → 1.2.57__tar.gz

{maisaedu_utilities_prefect-1.2.55 → maisaedu_utilities_prefect-1.2.57}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maisaedu_utilities_prefect
-Version: 1.2.55
+Version: 1.2.57
 Summary: Utilities for interaction with Prefect, for +A Education
 Home-page: UNKNOWN
 Author: A+ Educação

{maisaedu_utilities_prefect-1.2.55 → maisaedu_utilities_prefect-1.2.57}/maisaedu_utilities_prefect/llm/service/llm_service.py

@@ -82,27 +82,28 @@ class LLMService(BaseModel):
         ):
             raise ValueError("messages must be a list of dicts with 'role' and 'content' (str)")
 
-    def __save_history(self, config_id, request, response):
+    def __save_history(self, config_id, request, response, external_reference_id, metadata):
         conn = connect(get_dsn())
         model = request.get("model", "")
         request_json = json.dumps(request).replace("\\u0000", "")
         response_json = json.dumps(response.json()).replace("\\u0000", "")
-    
+        metadata_json = json.dumps(metadata).replace("\\u0000", "")
+
         insert_sql = f"""
-            INSERT INTO {self.default_schema}.llm_flows_runs_history (llm_flow_config_id, model, request, response)
-            VALUES (%s, %s, %s, %s)
+            INSERT INTO {self.default_schema}.llm_flows_runs_history (llm_flow_config_id, model, request, response, external_reference_id, metadata)
+            VALUES (%s, %s, %s, %s, %s, %s)
             RETURNING id;
         """
         
         with conn.cursor() as cur:
-            cur.execute(insert_sql, (config_id , model, request_json, response_json))
+            cur.execute(insert_sql, (config_id , model, request_json, response_json, external_reference_id, metadata_json))
             inserted_id = cur.fetchone()[0]
             conn.commit()
 
         conn.close()
         return inserted_id
 
-    def call_llm_api(self, messages):
+    def call_llm_api(self, messages, external_reference_id = None, metadata = None):
         self.__validate_messages(messages)
 
         results = []
@@ -145,7 +146,7 @@ class LLMService(BaseModel):
                     json=request,
                 )
 
-                llm_flow_run_history_id = self.__save_history(config['id'], request, response)
+                llm_flow_run_history_id = self.__save_history(config['id'], request, response, external_reference_id, metadata)
                 result_return["llm_flow_run_history_id"] = llm_flow_run_history_id
                 result_return["system_prompt"] = config['system_prompt']
                 result_return["response_format"] = config['response_format']

{maisaedu_utilities_prefect-1.2.55 → maisaedu_utilities_prefect-1.2.57}/maisaedu_utilities_prefect/powerbi_api/services/userService.py

@@ -1,7 +1,6 @@
-import json
 import requests
-import urllib
 import warnings
+from retry import retry
 
 from requests.exceptions import HTTPError
 
@@ -11,6 +10,7 @@ class UserService:
         self.client = client
         self.base_url = f"{self.client.api_url}/{self.client.api_version_snippet}/{self.client.api_myorg_snippet}/"
 
+    @retry(tries=5, delay=1, jitter=1)
     def refresh_user_permissions(self):
         """
         Refreshes user permissions in Power BI.

{maisaedu_utilities_prefect-1.2.55 → maisaedu_utilities_prefect-1.2.57}/maisaedu_utilities_prefect.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maisaedu-utilities-prefect
-Version: 1.2.55
+Version: 1.2.57
 Summary: Utilities for interaction with Prefect, for +A Education
 Home-page: UNKNOWN
 Author: A+ Educação

{maisaedu_utilities_prefect-1.2.55 → maisaedu_utilities_prefect-1.2.57}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="maisaedu_utilities_prefect",
-    version="1.2.55",
+    version="1.2.57",
     description="Utilities for interaction with Prefect, for +A Education",
     license="MIT License",
     author="A+ Educação",