chore: 🚚 rename internet_research to web_search [EXEC-470]

AI21Labs · Dec 10, 2024 · 4199de4 · 4199de4
1 parent 3bdad90
commit 4199de4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ai21/models/assistant/assistant.py b/ai21/models/assistant/assistant.py
@@ -3,10 +3,10 @@
 from typing_extensions import TypedDict
 
 Optimization = Literal["cost", "latency"]
-Tool = Literal["rag", "internet_research", "plan_approval"]
+Tool = Literal["rag", "web_search", "plan_approval"]
 
 
 class ToolResources(TypedDict, total=False):
     rag: Optional[dict]
-    internet_research: Optional[dict]
+    web_search: Optional[dict]
     plan_approval: Optional[dict]