fix: Patch session by merging metadata and replacinf all other values (…

…#202)
julep-ai · Apr 17, 2024 · 0dcfd17 · 0dcfd17
1 parent 6a912e1
commit 0dcfd17
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/agents-api/agents_api/activities/types.py b/agents-api/agents_api/activities/types.py
@@ -25,10 +25,12 @@ class ChatML(BaseModel):
     token_count: Optional[int] = None
 
 
-class BaseTask(BaseModel): ...
+class BaseTask(BaseModel):
+    ...
 
 
-class BaseTaskArgs(BaseModel): ...
+class BaseTaskArgs(BaseModel):
+    ...
 
 
 class AddPrinciplesTaskArgs(BaseTaskArgs):

diff --git a/agents-api/agents_api/clients/worker/types.py b/agents-api/agents_api/clients/worker/types.py
@@ -25,10 +25,12 @@ class ChatML(BaseModel):
     token_count: Optional[int] = None
 
 
-class BaseTask(BaseModel): ...
+class BaseTask(BaseModel):
+    ...
 
 
-class BaseTaskArgs(BaseModel): ...
+class BaseTaskArgs(BaseModel):
+    ...
 
 
 class MemoryManagementTaskArgs(BaseTaskArgs):

diff --git a/agents-api/agents_api/models/agent/create_agent.py b/agents-api/agents_api/models/agent/create_agent.py
@@ -42,7 +42,6 @@ def create_agent_query(
     default_settings: dict = {},
     client: CozoClient = client,
 ) -> pd.DataFrame:
-
     settings_cols, settings_vals = cozo_process_mutate_data(
         {
             **default_settings,