Skip to content

Commit 6524d22

Browse files
authored
Merge pull request #15 from Azure-Samples/githubmodels
Make github models the default
2 parents 8096eef + 87c8057 commit 6524d22

File tree

6 files changed

+15
-18
lines changed

6 files changed

+15
-18
lines changed

extract_github_issue.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
load_dotenv()
1414

1515

16-
if os.getenv("OPENAI_HOST", "azure") == "azure":
16+
if os.getenv("OPENAI_HOST", "github") == "azure":
1717
if not os.getenv("AZURE_OPENAI_SERVICE") or not os.getenv("AZURE_OPENAI_GPT_DEPLOYMENT"):
1818
logging.warning("AZURE_OPENAI_SERVICE and AZURE_OPENAI_GPT_DEPLOYMENT env variables are empty. See README.")
1919
exit(1)
@@ -54,6 +54,7 @@ class Issue(BaseModel):
5454
type: IssueType
5555
operating_system: str
5656

57+
5758
# Fetch an issue from a public GitHub repository
5859
url = "https://api.github.com/repos/Azure-Samples/azure-search-openai-demo/issues/2231"
5960
response = requests.get(url)
@@ -73,8 +74,7 @@ class Issue(BaseModel):
7374
)
7475

7576
message = completion.choices[0].message
76-
if (message.refusal):
77+
if message.refusal:
7778
print(message.refusal)
7879
else:
7980
print(message.parsed)
80-

extract_github_repo.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,14 @@
66
import azure.identity
77
import openai
88
import requests
9-
from rich import print
109
from dotenv import load_dotenv
1110
from pydantic import BaseModel, Field
11+
from rich import print
1212

1313
logging.basicConfig(level=logging.WARNING)
1414
load_dotenv()
1515

16-
if os.getenv("OPENAI_HOST", "azure") == "azure":
16+
if os.getenv("OPENAI_HOST", "github") == "azure":
1717
if not os.getenv("AZURE_OPENAI_SERVICE") or not os.getenv("AZURE_OPENAI_GPT_DEPLOYMENT"):
1818
logging.warning("AZURE_OPENAI_SERVICE and AZURE_OPENAI_GPT_DEPLOYMENT env variables are empty. See README.")
1919
exit(1)
@@ -95,7 +95,7 @@ class RepoOverview(BaseModel):
9595
)
9696

9797
message = completion.choices[0].message
98-
if (message.refusal):
98+
if message.refusal:
9999
print(message.refusal)
100100
else:
101-
print(message.parsed)
101+
print(message.parsed)

extract_image_graph.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
logging.basicConfig(level=logging.WARNING)
1212
load_dotenv()
1313

14-
if os.getenv("OPENAI_HOST", "azure") == "azure":
14+
if os.getenv("OPENAI_HOST", "github") == "azure":
1515
if not os.getenv("AZURE_OPENAI_SERVICE") or not os.getenv("AZURE_OPENAI_GPT_DEPLOYMENT"):
1616
logging.warning("AZURE_OPENAI_SERVICE and AZURE_OPENAI_GPT_DEPLOYMENT env variables are empty. See README.")
1717
exit(1)
@@ -73,8 +73,7 @@ def open_image_as_base64(filename):
7373
)
7474

7575
message = completion.choices[0].message
76-
if (message.refusal):
76+
if message.refusal:
7777
print(message.refusal)
7878
else:
7979
print(message.parsed)
80-

extract_image_table.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
logging.basicConfig(level=logging.WARNING)
1212
load_dotenv()
1313

14-
if os.getenv("OPENAI_HOST", "azure") == "azure":
14+
if os.getenv("OPENAI_HOST", "github") == "azure":
1515
if not os.getenv("AZURE_OPENAI_SERVICE") or not os.getenv("AZURE_OPENAI_GPT_DEPLOYMENT"):
1616
logging.warning("AZURE_OPENAI_SERVICE and AZURE_OPENAI_GPT_DEPLOYMENT env variables are empty. See README.")
1717
exit(1)
@@ -81,7 +81,7 @@ def open_image_as_base64(filename):
8181
)
8282

8383
message = completion.choices[0].message
84-
if (message.refusal):
84+
if message.refusal:
8585
print(message.refusal)
8686
else:
8787
print(message.parsed)

extract_pdf_receipt.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
logging.basicConfig(level=logging.WARNING)
1212
load_dotenv()
1313

14-
if os.getenv("OPENAI_HOST", "azure") == "azure":
14+
if os.getenv("OPENAI_HOST", "github") == "azure":
1515
if not os.getenv("AZURE_OPENAI_SERVICE") or not os.getenv("AZURE_OPENAI_GPT_DEPLOYMENT"):
1616
logging.warning("AZURE_OPENAI_SERVICE and AZURE_OPENAI_GPT_DEPLOYMENT env variables are empty. See README.")
1717
exit(1)
@@ -67,8 +67,7 @@ class Receipt(BaseModel):
6767
)
6868

6969
message = completion.choices[0].message
70-
if (message.refusal):
70+
if message.refusal:
7171
print(message.refusal)
7272
else:
7373
print(message.parsed)
74-

extract_webpage.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
logging.basicConfig(level=logging.WARNING)
1313
load_dotenv()
1414

15-
if os.getenv("OPENAI_HOST", "azure") == "azure":
15+
if os.getenv("OPENAI_HOST", "github") == "azure":
1616
if not os.getenv("AZURE_OPENAI_SERVICE") or not os.getenv("AZURE_OPENAI_GPT_DEPLOYMENT"):
1717
logging.warning("AZURE_OPENAI_SERVICE and AZURE_OPENAI_GPT_DEPLOYMENT env variables are empty. See README.")
1818
exit(1)
@@ -68,8 +68,7 @@ class BlogPost(BaseModel):
6868
)
6969

7070
message = completion.choices[0].message
71-
if (message.refusal):
71+
if message.refusal:
7272
print(message.refusal)
7373
else:
7474
print(message.parsed)
75-

0 commit comments

Comments
 (0)