chore(examples): update provider names to match tokens dictionary

f-aguzzi · f-aguzzi · commit ee078cb102ad · 2024-08-13T19:33:29.000+02:00
diff --git a/examples/azure/csv_scraper_azure.py b/examples/azure/csv_scraper_azure.py
@@ -25,7 +25,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/csv_scraper_graph_multi_azure.py b/examples/azure/csv_scraper_graph_multi_azure.py
@@ -25,7 +25,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/json_scraper_azure.py b/examples/azure/json_scraper_azure.py
@@ -23,7 +23,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/json_scraper_multi_azure.py b/examples/azure/json_scraper_multi_azure.py
@@ -12,7 +12,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/pdf_scraper_azure.py b/examples/azure/pdf_scraper_azure.py
@@ -10,7 +10,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/scrape_plain_text_azure.py b/examples/azure/scrape_plain_text_azure.py
@@ -28,7 +28,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/script_generator_azure.py b/examples/azure/script_generator_azure.py
@@ -15,7 +15,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/script_multi_generator_azure.py b/examples/azure/script_multi_generator_azure.py
@@ -16,7 +16,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/search_graph_azure.py b/examples/azure/search_graph_azure.py
@@ -22,7 +22,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/search_graph_schema_azure.py b/examples/azure/search_graph_schema_azure.py
@@ -30,7 +30,7 @@ class Dishes(BaseModel):
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/search_link_graph_azure.py b/examples/azure/search_link_graph_azure.py
@@ -15,7 +15,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/smart_scraper_azure.py b/examples/azure/smart_scraper_azure.py
@@ -26,7 +26,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/smart_scraper_multi_azure.py b/examples/azure/smart_scraper_multi_azure.py
@@ -14,7 +14,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/smart_scraper_schema_azure.py b/examples/azure/smart_scraper_schema_azure.py
@@ -28,7 +28,7 @@ class Projects(BaseModel):
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/xml_scraper_azure.py b/examples/azure/xml_scraper_azure.py
@@ -24,7 +24,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/azure/xml_scraper_graph_multi_azure.py b/examples/azure/xml_scraper_graph_multi_azure.py
@@ -25,7 +25,7 @@
 graph_config = {
     "llm": {
         "api_key": os.environ["AZURE_OPENAI_KEY"],
-        "model": "azure/gpt-3.5-turbo",
+        "model": "azure_openai/gpt-3.5-turbo",
     },
     "verbose": True,
     "headless": False
diff --git a/examples/gemini/csv_scraper_gemini.py b/examples/gemini/csv_scraper_gemini.py
@@ -24,7 +24,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
diff --git a/examples/gemini/csv_scraper_graph_multi_gemini.py b/examples/gemini/csv_scraper_graph_multi_gemini.py
@@ -28,7 +28,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
diff --git a/examples/gemini/custom_graph_gemini.py b/examples/gemini/custom_graph_gemini.py
@@ -18,7 +18,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
         "temperature": 0,
         "streaming": True
     },
diff --git a/examples/gemini/json_scraper_gemini.py b/examples/gemini/json_scraper_gemini.py
@@ -28,7 +28,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
diff --git a/examples/gemini/json_scraper_multi_gemini.py b/examples/gemini/json_scraper_multi_gemini.py
@@ -13,7 +13,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
     "library": "beautifulsoup"
 }
diff --git a/examples/gemini/pdf_scraper_graph_gemini.py b/examples/gemini/pdf_scraper_graph_gemini.py
@@ -19,7 +19,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
diff --git a/examples/gemini/pdf_scraper_multi_gemini.py b/examples/gemini/pdf_scraper_multi_gemini.py
@@ -13,7 +13,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
     "library": "beautifulsoup"
 }
diff --git a/examples/gemini/scrape_plain_text_gemini.py b/examples/gemini/scrape_plain_text_gemini.py
@@ -29,7 +29,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
         "temperature": 0,
         "streaming": True
     },
diff --git a/examples/gemini/scrape_xml_gemini.py b/examples/gemini/scrape_xml_gemini.py
@@ -29,7 +29,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
         "temperature": 0,
         "streaming": True
     },
diff --git a/examples/gemini/script_generator_gemini.py b/examples/gemini/script_generator_gemini.py
@@ -19,7 +19,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
     "library": "beautifoulsoup"
 }
diff --git a/examples/gemini/script_multi_generator_gemini.py b/examples/gemini/script_multi_generator_gemini.py
@@ -18,7 +18,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
     "library": "beautifoulsoup"
 }
diff --git a/examples/gemini/search_graph_gemini.py b/examples/gemini/search_graph_gemini.py
@@ -17,7 +17,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
         "temperature": 0,
         "streaming": True
     },
diff --git a/examples/gemini/search_graph_schema_gemini.py b/examples/gemini/search_graph_schema_gemini.py
@@ -32,7 +32,7 @@ class Dishes(BaseModel):
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
diff --git a/examples/gemini/search_link_graph_gemini.py b/examples/gemini/search_link_graph_gemini.py
@@ -17,7 +17,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 # ************************************************
diff --git a/examples/gemini/smart_scraper_gemini.py b/examples/gemini/smart_scraper_gemini.py
@@ -18,7 +18,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
diff --git a/examples/gemini/smart_scraper_multi_gemini.py b/examples/gemini/smart_scraper_multi_gemini.py
@@ -17,7 +17,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
diff --git a/examples/gemini/smart_scraper_schema_gemini.py b/examples/gemini/smart_scraper_schema_gemini.py
@@ -29,7 +29,7 @@ class Projects(BaseModel):
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 
@@ -54,4 +54,3 @@ class Projects(BaseModel):
 
 graph_exec_info = smart_scraper_graph.get_execution_info()
 print(prettify_exec_info(graph_exec_info))
-```
diff --git a/examples/gemini/xml_scraper_gemini.py b/examples/gemini/xml_scraper_gemini.py
@@ -28,7 +28,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 # ************************************************
diff --git a/examples/gemini/xml_scraper_graph_multi_gemini.py b/examples/gemini/xml_scraper_graph_multi_gemini.py
@@ -28,7 +28,7 @@
 graph_config = {
     "llm": {
         "api_key": gemini_key,
-        "model": "gemini-pro",
+        "model": "google_genai/gemini-pro",
     },
 }
 

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@`
`24`	`24`	`graph_config = {`
`25`	`25`	`"llm": {`
`26`	`26`	`"api_key": gemini_key,`
`27`		`- "model": "gemini-pro",`
	`27`	`+ "model": "google_genai/gemini-pro",`
`28`	`28`	`},`
`29`	`29`	`}`
`30`	`30`
Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@`
`28`	`28`	`graph_config = {`
`29`	`29`	`"llm": {`
`30`	`30`	`"api_key": gemini_key,`
`31`		`- "model": "gemini-pro",`
	`31`	`+ "model": "google_genai/gemini-pro",`
`32`	`32`	`},`
`33`	`33`	`}`
`34`	`34`
Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@`
`13`	`13`	`graph_config = {`
`14`	`14`	`"llm": {`
`15`	`15`	`"api_key": gemini_key,`
`16`		`- "model": "gemini-pro",`
	`16`	`+ "model": "google_genai/gemini-pro",`
`17`	`17`	`},`
`18`	`18`	`"library": "beautifulsoup"`
`19`	`19`	`}`
Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@`
`19`	`19`	`graph_config = {`
`20`	`20`	`"llm": {`
`21`	`21`	`"api_key": gemini_key,`
`22`		`- "model": "gemini-pro",`
	`22`	`+ "model": "google_genai/gemini-pro",`
`23`	`23`	`},`
`24`	`24`	`}`
`25`	`25`
Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@`
`18`	`18`	`graph_config = {`
`19`	`19`	`"llm": {`
`20`	`20`	`"api_key": gemini_key,`
`21`		`- "model": "gemini-pro",`
	`21`	`+ "model": "google_genai/gemini-pro",`
`22`	`22`	`},`
`23`	`23`	`"library": "beautifoulsoup"`
`24`	`24`	`}`
Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ class Dishes(BaseModel):`
`32`	`32`	`graph_config = {`
`33`	`33`	`"llm": {`
`34`	`34`	`"api_key": gemini_key,`
`35`		`- "model": "gemini-pro",`
	`35`	`+ "model": "google_genai/gemini-pro",`
`36`	`36`	`},`
`37`	`37`	`}`
`38`	`38`
Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@`
`17`	`17`	`graph_config = {`
`18`	`18`	`"llm": {`
`19`	`19`	`"api_key": gemini_key,`
`20`		`- "model": "gemini-pro",`
	`20`	`+ "model": "google_genai/gemini-pro",`
`21`	`21`	`},`
`22`	`22`	`}`
`23`	`23`	`# ************************************************`