fix(typespec,agents-api): Rename integration providers

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Sep 25, 2024 · d62a98e · d62a98e
1 parent d8072a0
commit d62a98e
Show file tree

Hide file tree

Showing 4 changed files with 57 additions and 19 deletions.
diff --git a/agents-api/agents_api/autogen/Tools.py b/agents-api/agents_api/autogen/Tools.py
@@ -91,14 +91,11 @@ class IntegrationDef(BaseModel):
     )
     provider: Literal[
         "dummy",
-        "dall-e",
-        "duckduckgo",
-        "hackernews",
+        "dalle_image_generator",
+        "duckduckgo_search",
+        "hacker_news",
         "weather",
         "wikipedia",
-        "twitter",
-        "webpage",
-        "requests",
     ]
     """
     The provider of the integration
@@ -132,14 +129,11 @@ class IntegrationDefUpdate(BaseModel):
     provider: (
         Literal[
             "dummy",
-            "dall-e",
-            "duckduckgo",
-            "hackernews",
+            "dalle_image_generator",
+            "duckduckgo_search",
+            "hacker_news",
             "weather",
             "wikipedia",
-            "twitter",
-            "webpage",
-            "requests",
         ]
         | None
     ) = None

diff --git a/agents-api/migrations/migrate_1727235852_add_forward_tool_calls_option.py b/agents-api/migrations/migrate_1727235852_add_forward_tool_calls_option.py
@@ -1,4 +1,4 @@
-#/usr/bin/env python3
+# /usr/bin/env python3
 
 MIGRATION_ID = "add_forward_tool_calls_option"
 CREATED_AT = 1727235852.744035

diff --git a/agents-api/tests/sample_tasks/integration_example.yaml b/agents-api/tests/sample_tasks/integration_example.yaml
@@ -0,0 +1,44 @@
+name: Simple multi step task
+
+input_schema:
+  type: object
+  properties:
+    topics:
+      type: array
+      items:
+        type: string
+
+tools:
+  - type: function
+    function:
+      name: generate_questions
+      description: Generate a list of questions for a given topic
+      parameters:
+      type: object
+      properties:
+        topic:
+            type: string
+            description: The topic to generate questions for
+
+  - type: integration
+    name: duckduckgo_search
+    integration:
+      provider: duckduckgo
+      setup:
+        api_key: <something>
+      arguments:
+        language: en-US
+
+main:
+  - foreach:
+      in: _["topics"]
+      do:
+        prompt:
+          - role: system
+            content: |-
+              Generate a list of 10 questions for the topic {{_}} as valid yaml.
+        unwrap: true
+
+  - tool: duckduckgo_search
+    arguments:
+      query: "'\n'.join(_)"
diff --git a/typespec/common/scalars.tsp b/typespec/common/scalars.tsp
@@ -57,12 +57,12 @@ scalar JinjaTemplate extends string;
 /** Integration provider name */
 alias integrationProvider = (
     | "dummy"
-    | "dall-e"
-    | "duckduckgo"
-    | "hackernews"
+    | "dalle_image_generator"
+    | "duckduckgo_search"
+    | "hacker_news"
     | "weather"
     | "wikipedia"
-    | "twitter"
-    | "webpage"
-    | "requests"
+    // | "twitter"
+    // | "webpage"
+    // | "requests"
 );