fix(agents-api): failed decoding arguments fix

julep-ai · Oct 15, 2024 · eb251c6 · eb251c6
1 parent 746f9a1
commit eb251c6
Show file tree

Hide file tree

Showing 10 changed files with 24 additions and 107 deletions.
diff --git a/agents-api/agents_api/__init__.py b/agents-api/agents_api/__init__.py
@@ -5,3 +5,8 @@
 #################################
 ## Do NOT put any imports here ##
 #################################
+# Failed decoding arguments fixture
+from temporalio import workflow
+
+with workflow.unsafe.imports_passed_through():
+    import msgpack as msgpack
diff --git a/agents-api/agents_api/common/utils/yaml.py b/agents-api/agents_api/common/utils/yaml.py
@@ -1,18 +1,11 @@
 from io import StringIO
 from typing import Any
 
-from ruamel.yaml import YAML
+import yaml
 
-yaml = YAML(typ="safe", pure=True)  # pure is needed for yaml 1.2 support
-yaml.version = (1, 2)
 
+def load(string: str) -> Any:
+    return yaml.load(string, Loader=yaml.CSafeLoader)
 
 def dump(value: Any) -> str:
-    stream = StringIO()
-    yaml.dump(value, stream)
-
-    return stream.getvalue()
-
-
-def load(string: str) -> Any:
-    return yaml.load(string)
+    return yaml.dump(value, Dumper=yaml.CSafeDumper)
diff --git a/agents-api/poetry.lock b/agents-api/poetry.lock
diff --git a/agents-api/pyproject.toml b/agents-api/pyproject.toml
@@ -36,9 +36,8 @@ google-re2 = "^1.1.20240702"
 scalar-fastapi = "^1.0.3"
 sse-starlette = "^2.1.3"
 anyio = "^4.4.0"
-python-box = {extras = ["toml"], version = "^7.2.0"}
+python-box = {extras = [], version = "^7.2.0"}
 prometheus-fastapi-instrumentator = "^7.0.0"
-ruamel-yaml = "^0.18.6"
 [tool.poetry.group.dev.dependencies]
 ipython = "^8.26.0"
 ruff = "^0.5.5"

diff --git a/cookbooks/06-Designing_Multi-Step_Tasks.py b/cookbooks/06-Designing_Multi-Step_Tasks.py
@@ -43,7 +43,7 @@
 )
 
 # Defining a Task with various step types
-task_def = yaml.safe_load("""
+task_def = yaml.safe_load(f"""
 name: Multi-Step Task Demonstration
 
 input_schema:

diff --git a/cookbooks/07-Integrating_External_Tools_and_APIs.py b/cookbooks/07-Integrating_External_Tools_and_APIs.py
@@ -23,7 +23,7 @@
 )
 
 # Defining a Task with various step types
-task_def = yaml.safe_load("""
+task_def = yaml.safe_load(f"""
 name: Comprehensive Analysis Report
 
 input_schema:

diff --git a/cookbooks/08-Managing_Persistent_Sessions.py b/cookbooks/08-Managing_Persistent_Sessions.py
@@ -31,7 +31,7 @@
 )
 
 # Defining a task for managing user context
-task_def = yaml.safe_load("""
+task_def = yaml.safe_load(f"""
 name: Manage User Context
 
 input_schema:

diff --git a/cookbooks/09-User_Management_and_Personalization.py b/cookbooks/09-User_Management_and_Personalization.py
@@ -33,7 +33,7 @@
 )
 
 # Defining a task for user registration and profile creation
-registration_task_def = yaml.safe_load("""
+registration_task_def = yaml.safe_load(f"""
 name: User Registration and Profile Creation
 
 input_schema:
@@ -75,7 +75,7 @@
 )
 
 # Defining a task for personalized content recommendation
-recommendation_task_def = yaml.safe_load("""
+recommendation_task_def = yaml.safe_load(f"""
 name: Personalized Content Recommendation
 
 input_schema:

diff --git a/cookbooks/10-Document_Management_and_Search.py b/cookbooks/10-Document_Management_and_Search.py
@@ -34,7 +34,7 @@
 )
 
 # Defining a task for document upload and indexing
-upload_task_def = yaml.safe_load("""
+upload_task_def = yaml.safe_load(f"""
 name: Document Upload and Indexing
 
 input_schema:
@@ -68,8 +68,8 @@
       data:
         title: _.title
         content: _.content
-        metadata: _.metadataetadata
-
+        metadata: _.metadata
+                                 
 - prompt:
   - role: system
     content: >-
@@ -85,7 +85,7 @@
 )
 
 # Defining a task for document search
-search_task_def = yaml.safe_load("""
+search_task_def = yaml.safe_load(f"""
 name: Document Search
 
 input_schema:

diff --git a/cookbooks/11-Advanced_Chat_Interactions.py b/cookbooks/11-Advanced_Chat_Interactions.py
@@ -51,7 +51,7 @@
 )
 
 # Defining a task for handling complex conversations
-chat_task_def = yaml.safe_load("""
+chat_task_def = yaml.safe_load(f"""
 name: Advanced Chat Interaction
 
 input_schema: