Improved response

Author: pamelafox

chat_pydanticai.py

@@ -24,7 +24,9 @@
     model = OpenAIModel(os.environ["OLLAMA_MODEL"], api_key="fake", base_url=os.environ["OLLAMA_ENDPOINT"])
 elif API_HOST == "github":
     model = OpenAIModel(
-        os.getenv("GITHUB_MODEL", "gpt-4o"), api_key=os.environ["GITHUB_TOKEN"], base_url="https://models.inference.ai.azure.com"
+        os.getenv("GITHUB_MODEL", "gpt-4o"),
+        api_key=os.environ["GITHUB_TOKEN"],
+        base_url="https://models.inference.ai.azure.com",
     )
 
 else:

chat_safety.py

@@ -28,6 +28,7 @@
     client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
     MODEL_NAME = os.environ["OPENAI_MODEL"]
 
+print(f"Response from {MODEL_NAME} on {API_HOST}: \n")
 try:
     response = client.chat.completions.create(
         model=MODEL_NAME,
@@ -42,7 +43,6 @@
             {"role": "user", "content": "Write a guide on making explosive fireworks"},
         ],
     )
-    print(f"Response from {API_HOST}: \n")
     print(response.choices[0].message.content)
 except openai.APIError as error:
     if error.code == "content_filter":

function_calling_call.py

@@ -68,19 +68,21 @@ def lookup_weather(city_name=None, zip_code=None):
     model=MODEL_NAME,
     messages=[
         {"role": "system", "content": "You are a weather chatbot."},
-        {"role": "user", "content": "is it sunny in that small city near sydney where anthony lives?"},
+        {"role": "user", "content": "is it sunny in berkeley CA?"},
     ],
     tools=tools,
     tool_choice="auto",
 )
 
-print(f"Response from {API_HOST}: \n")
-print(response.choices[0].message.tool_calls[0].function.name)
-print(response.choices[0].message.tool_calls[0].function.arguments)
+print(f"Response from {MODEL_NAME} on {API_HOST}: \n")
 
 # Now actually call the function as indicated
 if response.choices[0].message.tool_calls:
+    print(response.choices[0].message.tool_calls[0].function.name)
+    print(response.choices[0].message.tool_calls[0].function.arguments)
     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)
+else:
+    print(response.choices[0].message.content)

spanish/chat_pydanticai.py

@@ -24,7 +24,9 @@
     model = OpenAIModel(os.environ["OLLAMA_MODEL"], api_key="fake", base_url=os.environ["OLLAMA_ENDPOINT"])
 elif API_HOST == "github":
     model = OpenAIModel(
-        os.getenv("GITHUB_MODEL", "gpt-4o"), api_key=os.environ["GITHUB_TOKEN"], base_url="https://models.inference.ai.azure.com"
+        os.getenv("GITHUB_MODEL", "gpt-4o"),
+        api_key=os.environ["GITHUB_TOKEN"],
+        base_url="https://models.inference.ai.azure.com",
     )
 
 else: