maisaedu-utilities-prefect 1.2.56__tar.gz → 1.2.58__tar.gz

{maisaedu_utilities_prefect-1.2.56 → maisaedu_utilities_prefect-1.2.58}/PKG-INFO

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

{maisaedu_utilities_prefect-1.2.56 → maisaedu_utilities_prefect-1.2.58}/maisaedu_utilities_prefect/llm/service/llm_service.py

@@ -82,27 +82,31 @@ 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", "")
-    
+        if metadata:
+            metadata_json = json.dumps(metadata).replace("\\u0000", "")
+        else:
+            metadata_json = None
+
         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 +149,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.56 → maisaedu_utilities_prefect-1.2.58}/maisaedu_utilities_prefect.egg-info/PKG-INFO

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

{maisaedu_utilities_prefect-1.2.56 → maisaedu_utilities_prefect-1.2.58}/setup.py

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