change the content field of message to a interface

Author: burnerlee

.gitignore

@@ -2,3 +2,4 @@ dist/
 __pycache__
 .pytest_cache/
 *.egg-info/
+.DS_Store

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "CompextAI"
-version = "0.0.13"
+version = "0.0.14"
 authors = [
   { name="burnerlee", email="[email protected]" },
 ]

src/compextAI/execution.py

@@ -37,13 +37,14 @@ class ExecuteMessagesResponse:
     def __init__(self, thread_execution_id:str):
         self.thread_execution_id = thread_execution_id
 
-def execute_messages(client:APIClient, thread_execution_param_id:str, messages:list[Message],system_prompt:str="", append_assistant_response:bool=True) -> ThreadExecutionResponse:
+def execute_messages(client:APIClient, thread_execution_param_id:str, messages:list[Message],system_prompt:str="", append_assistant_response:bool=True, metadata:dict={}) -> ThreadExecutionResponse:
     thread_id = "compext_thread_null"
     response = client.post(f"/thread/{thread_id}/execute", data={
             "thread_execution_param_id": thread_execution_param_id,
             "append_assistant_response": append_assistant_response,
             "thread_execution_system_prompt": system_prompt,
-            "messages": [message.to_dict() for message in messages]
+            "messages": [message.to_dict() for message in messages],
+            "metadata": metadata
         })
 
     status_code: int = response["status"]

src/compextAI/messages.py

@@ -4,13 +4,13 @@
 class Message:
     message_id: str
     thread_id: str
-    content: str
+    content: any
     role: str
     metadata: dict
     created_at: datetime
     updated_at: datetime
 
-    def __init__(self,content:str, role:str, message_id:str='', thread_id:str='', metadata:dict={}, created_at:datetime=None, updated_at:datetime=None):
+    def __init__(self,content:any, role:str, message_id:str='', thread_id:str='', metadata:dict={}, created_at:datetime=None, updated_at:datetime=None):
         self.message_id = message_id
         self.thread_id = thread_id
         self.content = content
@@ -69,7 +69,7 @@ def create(client:APIClient, thread_id:str, messages:list[Message]) -> list[Mess
 
     return [get_message_object_from_dict(message) for message in data]
 
-def update(client:APIClient, message_id:str, content:str, role:str, metadata:dict={}) -> Message:
+def update(client:APIClient, message_id:str, content:any, role:str, metadata:dict={}) -> Message:
     response = client.put(f"/message/{message_id}", data={"content": content, "role": role, "metadata": metadata})
 
     status_code: int = response["status"]

src/compextAI/params.py

@@ -68,8 +68,8 @@ def create(client:APIClient, name:str, environment:str, project_name:str, templa
 
     return get_thread_execution_param_object_from_dict(data)
 
-def update(client:APIClient, name:str, environment:str, model:str, temperature:float, timeout:int, max_tokens:int, max_completion_tokens:int, top_p:float, max_output_tokens:int, response_format:any, system_prompt:str) -> ThreadExecutionParam:
-    response = client.post(f"/execparams/update", data={"name": name, "environment": environment, "model": model, "temperature": temperature, "timeout": timeout, "max_tokens": max_tokens, "max_completion_tokens": max_completion_tokens, "top_p": top_p, "max_output_tokens": max_output_tokens, "response_format": response_format, "system_prompt": system_prompt})
+def update(client:APIClient, name:str, environment:str, templateID:str) -> ThreadExecutionParam:
+    response = client.post(f"/execparams/update", data={"name": name, "environment": environment, "template_id": templateID})
 
     status_code: int = response["status"]
     data: dict = response["data"]

src/compextAI/threads.py

@@ -21,11 +21,12 @@ def __init__(self, thread_id:str, title:str, metadata:dict):
     def __str__(self):
         return f"Thread(thread_id={self.thread_id}, title={self.title}, metadata={self.metadata})"
 
-    def execute(self, client:APIClient, thread_exec_param_id: str, system_prompt:str="", append_assistant_response:bool=True) -> ThreadExecutionResponse:
+    def execute(self, client:APIClient, thread_exec_param_id: str, system_prompt:str="", append_assistant_response:bool=True, metadata:dict={}) -> ThreadExecutionResponse:
         response = client.post(f"/thread/{self.thread_id}/execute", data={
             "thread_execution_param_id": thread_exec_param_id,
             "append_assistant_response": append_assistant_response,
-            "thread_execution_system_prompt": system_prompt
+            "thread_execution_system_prompt": system_prompt,
+            "metadata": metadata
         })
 
         status_code: int = response["status"]