chore: 🚚 rename internet_research to web_search (#241)

EXEC-470 Co-authored-by: benshuk <[email protected]>
AI21Labs · Dec 11, 2024 · 770c6c5 · 770c6c5
1 parent 3bdad90
commit 770c6c5
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]