Skip to content

Commit b9dac76

Browse files
committed
Default to GitHub models
1 parent f370efd commit b9dac76

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

44 files changed

+88
-88
lines changed

chained_calls.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212

@@ -28,7 +28,7 @@
2828
elif API_HOST == "github":
2929

3030
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
31-
MODEL_NAME = os.environ["GITHUB_MODEL"]
31+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3232

3333
else:
3434

chat.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212

@@ -28,7 +28,7 @@
2828
elif API_HOST == "github":
2929

3030
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
31-
MODEL_NAME = os.environ["GITHUB_MODEL"]
31+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3232

3333
else:
3434

chat_async.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
99
load_dotenv(override=True)
10-
API_HOST = os.getenv("API_HOST")
10+
API_HOST = os.getenv("API_HOST", "github")
1111

1212
if API_HOST == "azure":
1313
token_provider = azure.identity.get_bearer_token_provider(
@@ -24,7 +24,7 @@
2424
MODEL_NAME = os.environ["OLLAMA_MODEL"]
2525
elif API_HOST == "github":
2626
client = openai.AsyncOpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
27-
MODEL_NAME = os.environ["GITHUB_MODEL"]
27+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
2828
else:
2929
client = openai.AsyncOpenAI(api_key=os.environ["OPENAI_KEY"])
3030
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_history.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212
token_provider = azure.identity.get_bearer_token_provider(
@@ -23,7 +23,7 @@
2323
MODEL_NAME = os.environ["OLLAMA_MODEL"]
2424
elif API_HOST == "github":
2525
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
26-
MODEL_NAME = os.environ["GITHUB_MODEL"]
26+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
2727
else:
2828
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
2929
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_history_stream.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212
token_provider = azure.identity.get_bearer_token_provider(
@@ -23,7 +23,7 @@
2323
MODEL_NAME = os.environ["OLLAMA_MODEL"]
2424
elif API_HOST == "github":
2525
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
26-
MODEL_NAME = os.environ["GITHUB_MODEL"]
26+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
2727
else:
2828
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
2929
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_langchain.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
99
load_dotenv(override=True)
10-
API_HOST = os.getenv("API_HOST")
10+
API_HOST = os.getenv("API_HOST", "github")
1111

1212
if API_HOST == "azure":
1313
token_provider = azure.identity.get_bearer_token_provider(
@@ -27,7 +27,7 @@
2727
)
2828
elif API_HOST == "github":
2929
llm = ChatOpenAI(
30-
model_name=os.environ["GITHUB_MODEL"],
30+
model_name=os.getenv("GITHUB_MODEL", "gpt-4o"),
3131
openai_api_base="https://models.inference.ai.azure.com",
3232
openai_api_key=os.environ["GITHUB_TOKEN"],
3333
)

chat_llamaindex.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1111
load_dotenv(override=True)
12-
API_HOST = os.getenv("API_HOST")
12+
API_HOST = os.getenv("API_HOST", "github")
1313

1414
if API_HOST == "azure":
1515
token_provider = azure.identity.get_bearer_token_provider(
@@ -29,7 +29,7 @@
2929
)
3030
elif API_HOST == "github":
3131
llm = OpenAILike(
32-
model=os.environ["GITHUB_MODEL"],
32+
model=os.getenv("GITHUB_MODEL", "gpt-4o"),
3333
api_base="https://models.inference.ai.azure.com",
3434
api_key=os.environ["GITHUB_TOKEN"],
3535
is_chat_model=True,

chat_pydanticai.py

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

99
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1010
load_dotenv(override=True)
11-
API_HOST = os.getenv("API_HOST")
11+
API_HOST = os.getenv("API_HOST", "github")
1212

1313
if API_HOST == "azure":
1414
token_provider = azure.identity.get_bearer_token_provider(
@@ -24,7 +24,7 @@
2424
model = OpenAIModel(os.environ["OLLAMA_MODEL"], api_key="fake", base_url=os.environ["OLLAMA_ENDPOINT"])
2525
elif API_HOST == "github":
2626
model = OpenAIModel(
27-
os.environ["GITHUB_MODEL"], api_key=os.environ["GITHUB_TOKEN"], base_url="https://models.inference.ai.azure.com"
27+
os.getenv("GITHUB_MODEL", "gpt-4o"), api_key=os.environ["GITHUB_TOKEN"], base_url="https://models.inference.ai.azure.com"
2828
)
2929

3030
else:

chat_safety.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212
token_provider = azure.identity.get_bearer_token_provider(
@@ -23,7 +23,7 @@
2323
MODEL_NAME = os.environ["OLLAMA_MODEL"]
2424
elif API_HOST == "github":
2525
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
26-
MODEL_NAME = os.environ["GITHUB_MODEL"]
26+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
2727
else:
2828
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
2929
MODEL_NAME = os.environ["OPENAI_MODEL"]

chat_stream.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212
token_provider = azure.identity.get_bearer_token_provider(
@@ -23,7 +23,7 @@
2323
MODEL_NAME = os.environ["OLLAMA_MODEL"]
2424
elif API_HOST == "github":
2525
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
26-
MODEL_NAME = os.environ["GITHUB_MODEL"]
26+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
2727
else:
2828
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
2929
MODEL_NAME = os.environ["OPENAI_MODEL"]

few_shot_examples.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212

@@ -28,7 +28,7 @@
2828
elif API_HOST == "github":
2929

3030
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
31-
MODEL_NAME = os.environ["GITHUB_MODEL"]
31+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3232

3333
else:
3434

function_calling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212

@@ -28,7 +28,7 @@
2828
elif API_HOST == "github":
2929

3030
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
31-
MODEL_NAME = os.environ["GITHUB_MODEL"]
31+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3232

3333
else:
3434

function_calling_call.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
99
load_dotenv(override=True)
10-
API_HOST = os.getenv("API_HOST")
10+
API_HOST = os.getenv("API_HOST", "github")
1111

1212
if API_HOST == "azure":
1313
token_provider = azure.identity.get_bearer_token_provider(
@@ -26,7 +26,7 @@
2626

2727
elif API_HOST == "github":
2828
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
29-
MODEL_NAME = os.environ["GITHUB_MODEL"]
29+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3030

3131
else:
3232
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

function_calling_multiple.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212

@@ -28,7 +28,7 @@
2828
elif API_HOST == "github":
2929

3030
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
31-
MODEL_NAME = os.environ["GITHUB_MODEL"]
31+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3232

3333
else:
3434

prompt_engineering.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
88
load_dotenv(override=True)
9-
API_HOST = os.getenv("API_HOST")
9+
API_HOST = os.getenv("API_HOST", "github")
1010

1111
if API_HOST == "azure":
1212

@@ -28,7 +28,7 @@
2828
elif API_HOST == "github":
2929

3030
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
31-
MODEL_NAME = os.environ["GITHUB_MODEL"]
31+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3232

3333
else:
3434

rag_csv.py

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

99
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1010
load_dotenv(override=True)
11-
API_HOST = os.getenv("API_HOST")
11+
API_HOST = os.getenv("API_HOST", "github")
1212

1313
if API_HOST == "azure":
1414
token_provider = azure.identity.get_bearer_token_provider(
@@ -27,7 +27,7 @@
2727

2828
elif API_HOST == "github":
2929
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
30-
MODEL_NAME = os.environ["GITHUB_MODEL"]
30+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3131

3232
else:
3333
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

rag_documents_flow.py

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

99
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1010
load_dotenv(override=True)
11-
API_HOST = os.getenv("API_HOST")
11+
API_HOST = os.getenv("API_HOST", "github")
1212

1313
if API_HOST == "azure":
1414
token_provider = azure.identity.get_bearer_token_provider(
@@ -27,7 +27,7 @@
2727

2828
elif API_HOST == "github":
2929
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
30-
MODEL_NAME = os.environ["GITHUB_MODEL"]
30+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3131

3232
else:
3333
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

rag_documents_hybrid.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1212
load_dotenv(override=True)
13-
API_HOST = os.getenv("API_HOST")
13+
API_HOST = os.getenv("API_HOST", "github")
1414

1515
if API_HOST == "azure":
1616
token_provider = azure.identity.get_bearer_token_provider(
@@ -29,7 +29,7 @@
2929

3030
elif API_HOST == "github":
3131
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
32-
MODEL_NAME = os.environ["GITHUB_MODEL"]
32+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3333

3434
else:
3535
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

rag_documents_ingestion.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1212
load_dotenv(override=True)
13-
API_HOST = os.getenv("API_HOST")
13+
API_HOST = os.getenv("API_HOST", "github")
1414

1515
if API_HOST == "azure":
1616
token_provider = azure.identity.get_bearer_token_provider(
@@ -29,7 +29,7 @@
2929

3030
elif API_HOST == "github":
3131
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
32-
MODEL_NAME = os.environ["GITHUB_MODEL"]
32+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3333

3434
else:
3535
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

rag_multiturn.py

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

99
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1010
load_dotenv(override=True)
11-
API_HOST = os.getenv("API_HOST")
11+
API_HOST = os.getenv("API_HOST", "github")
1212

1313
if API_HOST == "azure":
1414
token_provider = azure.identity.get_bearer_token_provider(
@@ -27,7 +27,7 @@
2727

2828
elif API_HOST == "github":
2929
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
30-
MODEL_NAME = os.environ["GITHUB_MODEL"]
30+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3131

3232
else:
3333
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

rag_queryrewrite.py

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

99
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
1010
load_dotenv(override=True)
11-
API_HOST = os.getenv("API_HOST")
11+
API_HOST = os.getenv("API_HOST", "github")
1212

1313
if API_HOST == "azure":
1414
token_provider = azure.identity.get_bearer_token_provider(
@@ -27,7 +27,7 @@
2727

2828
elif API_HOST == "github":
2929
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
30-
MODEL_NAME = os.environ["GITHUB_MODEL"]
30+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3131

3232
else:
3333
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

retrieval_augmented_generation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API
99
load_dotenv(override=True)
10-
API_HOST = os.getenv("API_HOST")
10+
API_HOST = os.getenv("API_HOST", "github")
1111

1212
if API_HOST == "azure":
1313
token_provider = azure.identity.get_bearer_token_provider(
@@ -26,7 +26,7 @@
2626

2727
elif API_HOST == "github":
2828
client = openai.OpenAI(base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
29-
MODEL_NAME = os.environ["GITHUB_MODEL"]
29+
MODEL_NAME = os.getenv("GITHUB_MODEL", "gpt-4o")
3030

3131
else:
3232
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

0 commit comments

Comments
 (0)