More function calling examples

Author: pamelafox

function_calling_call.py

@@ -1,5 +1,5 @@
-import os
 import json
+import os
 
 import azure.identity
 import openai
@@ -10,7 +10,6 @@
 API_HOST = os.getenv("API_HOST")
 
 if API_HOST == "azure":
-
     token_provider = azure.identity.get_bearer_token_provider(
         azure.identity.DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default"
     )
@@ -22,20 +21,17 @@
     MODEL_NAME = os.getenv("AZURE_OPENAI_DEPLOYMENT")
 
 elif API_HOST == "ollama":
-
     client = openai.OpenAI(
         base_url=os.getenv("OLLAMA_ENDPOINT"),
         api_key="nokeyneeded",
     )
     MODEL_NAME = os.getenv("OLLAMA_MODEL")
 
 elif API_HOST == "github":
-
     client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
     MODEL_NAME = os.getenv("GITHUB_MODEL")
 
 else:
-
     client = openai.OpenAI(api_key=os.getenv("OPENAI_KEY"))
     MODEL_NAME = os.getenv("OPENAI_MODEL")
 
@@ -45,6 +41,7 @@ def lookup_weather(city_name=None, zip_code=None):
     print(f"Looking up weather for {city_name or zip_code}...")
     return "It's sunny!"
 
+
 tools = [
     {
         "type": "function",
@@ -89,4 +86,4 @@ def lookup_weather(city_name=None, zip_code=None):
     function_name = response.choices[0].message.tool_calls[0].function.name
     arguments = json.loads(response.choices[0].message.tool_calls[0].function.arguments)
     if function_name == "lookup_weather":
-        lookup_weather(**arguments)
+        lookup_weather(**arguments)