feat(cookbooks): minor fixes

julep-ai · Oct 9, 2024 · 6bb5c0f · 6bb5c0f
1 parent 9894aa6
commit 6bb5c0f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 7 deletions.
diff --git a/cookbooks/11-Advanced_Chat_Interactions.py b/cookbooks/11-Advanced_Chat_Interactions.py
@@ -39,6 +39,17 @@
     model="gpt-4-turbo",
 )
 
+# Add a web search tool to the agent
+client.agents.tools.create(
+    agent_id=AGENT_UUID,
+    name="web_search",
+    integration={
+        "provider": "brave",
+        "method": "search",
+        "setup": {"api_key": "YOUR_BRAVE_API_KEY"},
+    },
+)
+
 # Defining a task for handling complex conversations
 chat_task_def = yaml.safe_load("""
 name: Advanced Chat Interaction

diff --git a/cookbooks/13-Error_Handling_and_Recovery.py b/cookbooks/13-Error_Handling_and_Recovery.py
@@ -75,11 +75,9 @@
           type: string
 
 main:
-- evaluate:
-    value: inputs[0].operation
-                          
+
 - switch:
-    case: "'divide'"
+    case: "inputs[0].operation == 'divide'"
       tool: divide
       arguments:
         divisor: inputs[0].value
@@ -89,7 +87,7 @@
           delay: 2
         fallback:
           return: "Error: Division by zero or invalid input"
-    case: "'api_call'"
+    case: "inputs[0].operation == 'api_call'"
         tool: api_call
         arguments:
           endpoint: "/status/{{inputs[0].value}}"
@@ -99,9 +97,9 @@
             delay: 5
           fallback:
             return: "Error: API call failed after multiple attempts"
-      case: "'process_data'"
+      case: "inputs[0].operation == 'process_data'"
         evaluate:
-            data: "'Sample data: ' + str(inputs[0].value)"
+          data: "'Sample data: ' + str(inputs[0].value)"
         tool: process_data
         arguments:
           data: _.data