Skip to content

Commit 923cb82

Browse files
committed
Update GitHub models
1 parent 913f705 commit 923cb82

14 files changed

+28
-28
lines changed

chained_calls.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030

3131
elif API_HOST == "github":
3232

33-
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
34-
MODEL_NAME = os.getenv("GITHUB_MODEL")
33+
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
34+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3535

3636
else:
3737

chat_async.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
)
2727
MODEL_NAME = os.getenv("OLLAMA_MODEL")
2828
elif API_HOST == "github":
29-
client = openai.AsyncOpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
30-
MODEL_NAME = os.getenv("GITHUB_MODEL")
29+
client = openai.AsyncOpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
30+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3131
else:
3232
client = openai.AsyncOpenAI(api_key=os.environ["OPENAI_KEY"])
3333
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_history.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@
2525
)
2626
MODEL_NAME = os.getenv("OLLAMA_MODEL")
2727
elif API_HOST == "github":
28-
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
29-
MODEL_NAME = os.getenv("GITHUB_MODEL")
28+
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
29+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3030
else:
3131
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
3232
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_history_stream.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@
2525
)
2626
MODEL_NAME = os.getenv("OLLAMA_MODEL")
2727
elif API_HOST == "github":
28-
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
29-
MODEL_NAME = os.getenv("GITHUB_MODEL")
28+
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
29+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3030
else:
3131
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
3232
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_langchain.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@
2727
)
2828
elif API_HOST == "github":
2929
llm = ChatOpenAI(
30-
model_name=os.getenv("GITHUB_MODEL"),
30+
model_name=os.environ["GITHUB_MODEL"],
3131
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"],
3333
)
3434
else:
3535
llm = ChatOpenAI(model_name=os.environ["OPENAI_MODEL"], openai_api_key=os.environ["OPENAI_KEY"])

chat_llamaindex.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@
2929
)
3030
elif API_HOST == "github":
3131
llm = OpenAILike(
32-
model=os.getenv("GITHUB_MODEL"),
32+
model=os.environ["GITHUB_MODEL"],
3333
api_base="https://models.inference.ai.azure.com",
34-
api_key=os.getenv("GITHUB_TOKEN"),
34+
api_key=os.environ["GITHUB_TOKEN"],
3535
is_chat_model=True,
3636
)
3737
else:

chat_safety.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@
2525
)
2626
MODEL_NAME = os.getenv("OLLAMA_MODEL")
2727
elif API_HOST == "github":
28-
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
29-
MODEL_NAME = os.getenv("GITHUB_MODEL")
28+
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
29+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3030
else:
3131
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
3232
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_stream.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@
2525
)
2626
MODEL_NAME = os.getenv("OLLAMA_MODEL")
2727
elif API_HOST == "github":
28-
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
29-
MODEL_NAME = os.getenv("GITHUB_MODEL")
28+
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
29+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3030
else:
3131
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
3232
MODEL_NAME = os.environ["OPENAI_MODEL"]

few_shot_examples.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030

3131
elif API_HOST == "github":
3232

33-
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
34-
MODEL_NAME = os.getenv("GITHUB_MODEL")
33+
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
34+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3535

3636
else:
3737

function_calling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030

3131
elif API_HOST == "github":
3232

33-
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.getenv("GITHUB_TOKEN"))
34-
MODEL_NAME = os.getenv("GITHUB_MODEL")
33+
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
34+
MODEL_NAME = os.environ["GITHUB_MODEL"]
3535

3636
else:
3737

0 commit comments

Comments
 (0)