Formatting

Author: pamelafox

chat.py

@@ -30,10 +30,7 @@
 
 elif API_HOST == "github":
 
-    client = openai.OpenAI(
-        base_url="https://models.inference.ai.azure.com",
-        api_key=os.getenv("GITHUB_TOKEN")
-    )
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
     MODEL_NAME = os.getenv("GITHUB_MODEL")
 
 else:

chat_async.py

@@ -26,10 +26,7 @@
     )
     MODEL_NAME = os.getenv("OLLAMA_MODEL")
 elif API_HOST == "github":
-    client = openai.AsyncOpenAI(
-        base_url="https://models.inference.ai.azure.com",
-        api_key=os.getenv("GITHUB_TOKEN")
-    )
+    client = openai.AsyncOpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
     MODEL_NAME = os.getenv("GITHUB_MODEL")
 else:
     client = openai.AsyncOpenAI(api_key=os.getenv("OPENAI_KEY"))

chat_history.py

@@ -25,10 +25,7 @@
     )
     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")
-    )
+    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"))

chat_history_stream.py

@@ -25,10 +25,7 @@
     )
     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")
-    )
+    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"))

chat_llamaindex.py

@@ -32,7 +32,8 @@
         model=os.getenv("GITHUB_MODEL"),
         api_base="https://models.inference.ai.azure.com",
         api_key=os.getenv("GITHUB_TOKEN"),
-        is_chat_model=True)
+        is_chat_model=True,
+    )
 else:
     llm = OpenAI(model=os.getenv("OPENAI_MODEL"), api_key=os.getenv("OPENAI_KEY"))
 

chat_safety.py

@@ -25,10 +25,7 @@
     )
     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")
-    )
+    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"))

chat_stream.py

@@ -25,10 +25,7 @@
     )
     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")
-    )
+    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"))