Update GitHub models

pamelafox · pamelafox · commit 923cb828dc5b · 2025-03-03T14:47:46.000-08:00
diff --git a/chained_calls.py b/chained_calls.py
@@ -30,8 +30,8 @@
 
 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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 
 else:
 
diff --git a/chat_async.py b/chat_async.py
@@ -26,8 +26,8 @@
     )
     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"))
-    MODEL_NAME = os.getenv("GITHUB_MODEL")
+    client = openai.AsyncOpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 else:
     client = openai.AsyncOpenAI(api_key=os.environ["OPENAI_KEY"])
     MODEL_NAME = os.environ["OPENAI_MODEL"]
diff --git a/chat_history.py b/chat_history.py
@@ -25,8 +25,8 @@
     )
     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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 else:
     client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
     MODEL_NAME = os.environ["OPENAI_MODEL"]
diff --git a/chat_history_stream.py b/chat_history_stream.py
@@ -25,8 +25,8 @@
     )
     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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 else:
     client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
     MODEL_NAME = os.environ["OPENAI_MODEL"]
diff --git a/chat_langchain.py b/chat_langchain.py
@@ -27,9 +27,9 @@
     )
 elif API_HOST == "github":
     llm = ChatOpenAI(
-        model_name=os.getenv("GITHUB_MODEL"),
+        model_name=os.environ["GITHUB_MODEL"],
         openai_api_base="https://models.inference.ai.azure.com",
-        openai_api_key=os.getenv("GITHUB_TOKEN"),
+        openai_api_key=os.environ["GITHUB_TOKEN"],
     )
 else:
     llm = ChatOpenAI(model_name=os.environ["OPENAI_MODEL"], openai_api_key=os.environ["OPENAI_KEY"])
diff --git a/chat_llamaindex.py b/chat_llamaindex.py
@@ -29,9 +29,9 @@
     )
 elif API_HOST == "github":
     llm = OpenAILike(
-        model=os.getenv("GITHUB_MODEL"),
+        model=os.environ["GITHUB_MODEL"],
         api_base="https://models.inference.ai.azure.com",
-        api_key=os.getenv("GITHUB_TOKEN"),
+        api_key=os.environ["GITHUB_TOKEN"],
         is_chat_model=True,
     )
 else:
diff --git a/chat_safety.py b/chat_safety.py
@@ -25,8 +25,8 @@
     )
     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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 else:
     client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
     MODEL_NAME = os.environ["OPENAI_MODEL"]
diff --git a/chat_stream.py b/chat_stream.py
@@ -25,8 +25,8 @@
     )
     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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 else:
     client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
     MODEL_NAME = os.environ["OPENAI_MODEL"]
diff --git a/few_shot_examples.py b/few_shot_examples.py
@@ -30,8 +30,8 @@
 
 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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 
 else:
 
diff --git a/function_calling.py b/function_calling.py
@@ -30,8 +30,8 @@
 
 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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 
 else:
 
diff --git a/function_calling_call.py b/function_calling_call.py
@@ -28,8 +28,8 @@
     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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 
 else:
     client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
diff --git a/function_calling_multiple.py b/function_calling_multiple.py
@@ -30,8 +30,8 @@
 
 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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 
 else:
 
diff --git a/prompt_engineering.py b/prompt_engineering.py
@@ -30,8 +30,8 @@
 
 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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 
 else:
 
diff --git a/retrieval_augmented_generation.py b/retrieval_augmented_generation.py
@@ -28,8 +28,8 @@
     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")
+    client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
+    MODEL_NAME = os.environ["GITHUB_MODEL"]
 
 else:
     client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

Original file line number	Diff line number	Diff line change
`@@ -27,9 +27,9 @@`
`27`	`27`	`)`
`28`	`28`	`elif API_HOST == "github":`
`29`	`29`	`llm = ChatOpenAI(`
`30`		`- model_name=os.getenv("GITHUB_MODEL"),`
	`30`	`+ model_name=os.environ["GITHUB_MODEL"],`
`31`	`31`	`openai_api_base="https://models.inference.ai.azure.com",`
`32`		`- openai_api_key=os.getenv("GITHUB_TOKEN"),`
	`32`	`+ openai_api_key=os.environ["GITHUB_TOKEN"],`
`33`	`33`	`)`
`34`	`34`	`else:`
`35`	`35`	`llm = ChatOpenAI(model_name=os.environ["OPENAI_MODEL"], openai_api_key=os.environ["OPENAI_KEY"])`
Original file line number	Diff line number	Diff line change
`@@ -29,9 +29,9 @@`
`29`	`29`	`)`
`30`	`30`	`elif API_HOST == "github":`
`31`	`31`	`llm = OpenAILike(`
`32`		`- model=os.getenv("GITHUB_MODEL"),`
	`32`	`+ model=os.environ["GITHUB_MODEL"],`
`33`	`33`	`api_base="https://models.inference.ai.azure.com",`
`34`		`- api_key=os.getenv("GITHUB_TOKEN"),`
	`34`	`+ api_key=os.environ["GITHUB_TOKEN"],`
`35`	`35`	`is_chat_model=True,`
`36`	`36`	`)`
`37`	`37`	`else:`