File tree Expand file tree Collapse file tree 13 files changed +3
-51
lines changed
Expand file tree Collapse file tree 13 files changed +3
-51
lines changed Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
Original file line number Diff line number Diff line change 99API_HOST = os .getenv ("API_HOST" , "github" )
1010
1111if API_HOST == "azure" :
12-
1312 token_provider = azure .identity .get_bearer_token_provider (
1413 azure .identity .DefaultAzureCredential (), "https://cognitiveservices.azure.com/.default"
1514 )
2120 MODEL_NAME = os .environ ["AZURE_OPENAI_DEPLOYMENT" ]
2221
2322elif API_HOST == "ollama" :
24-
2523 client = openai .OpenAI (base_url = os .environ ["OLLAMA_ENDPOINT" ], api_key = "nokeyneeded" )
2624 MODEL_NAME = os .environ ["OLLAMA_MODEL" ]
2725
2826elif API_HOST == "github" :
29-
3027 client = openai .OpenAI (base_url = "https://models.inference.ai.azure.com" , api_key = os .environ ["GITHUB_TOKEN" ])
3128 MODEL_NAME = os .getenv ("GITHUB_MODEL" , "gpt-4o" )
3229
3330else :
34-
3531 client = openai .OpenAI (api_key = os .environ ["OPENAI_KEY" ])
3632 MODEL_NAME = os .environ ["OPENAI_MODEL" ]
3733
7672 print (tool_call .function .name )
7773 print (tool_call .function .arguments )
7874else :
79- print (response .choices [0 ].message .content )
75+ print (response .choices [0 ].message .content )
You can’t perform that action at this time.
0 commit comments