mirror of
https://github.com/microsoft/semantic-kernel.git
synced 2026-10-12 11:19:56 +00:00
Jordan Henkel
0ce2acbf6d
This PR provides a path to using the ChatGPT API in the Python Preview of Semantic Kernel. In addition, this provides `Azure*` versions of many existing models (so Python users can leverage Azure OpenAI). I think that, in general, it may be worth considering how best to work with models that have different modalities: now we have text completions, embeddings, chat completions (and I'd imagine images/etc. may be nice to support someday too). Regardless, this PR provides a fast path to using the exciting new Chat APIs from OpenAI with SK! See the new `python/tests/chat_gpt_api.py` for a usage example.
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
from typing import Optional, Tuple
|
|
|
|
|
|
def openai_settings_from_dot_env() -> Tuple[str, Optional[str]]:
|
|
"""
|
|
Reads the OpenAI API key and organization ID from the .env file.
|
|
|
|
Returns:
|
|
Tuple[str, str]: The OpenAI API key, the OpenAI organization ID
|
|
"""
|
|
|
|
api_key, org_id = None, None
|
|
with open(".env", "r") as f:
|
|
lines = f.readlines()
|
|
|
|
for line in lines:
|
|
if line.startswith("OPENAI_API_KEY"):
|
|
parts = line.split("=")[1:]
|
|
api_key = "=".join(parts).strip().strip('"')
|
|
continue
|
|
|
|
if line.startswith("OPENAI_ORG_ID"):
|
|
parts = line.split("=")[1:]
|
|
org_id = "=".join(parts).strip().strip('"')
|
|
continue
|
|
|
|
assert api_key is not None, "OpenAI API key not found in .env file"
|
|
|
|
# It's okay if the org ID is not found (not required)
|
|
return api_key, org_id
|
|
|
|
|
|
def azure_openai_settings_from_dot_env(include_deployment=True) -> Tuple[str, str, str]:
|
|
"""
|
|
Reads the Azure OpenAI API key and endpoint from the .env file.
|
|
|
|
Returns:
|
|
Tuple[str, str, str]: The deployment name (or empty), Azure OpenAI API key,
|
|
and the endpoint
|
|
"""
|
|
|
|
deployment, api_key, endpoint = None, None, None
|
|
with open(".env", "r") as f:
|
|
lines = f.readlines()
|
|
|
|
for line in lines:
|
|
if include_deployment and line.startswith("AZURE_OPENAI_DEPLOYMENT_NAME"):
|
|
parts = line.split("=")[1:]
|
|
deployment = "=".join(parts).strip().strip('"')
|
|
continue
|
|
|
|
if line.startswith("AZURE_OPENAI_API_KEY"):
|
|
parts = line.split("=")[1:]
|
|
api_key = "=".join(parts).strip().strip('"')
|
|
continue
|
|
|
|
if line.startswith("AZURE_OPENAI_ENDPOINT"):
|
|
parts = line.split("=")[1:]
|
|
endpoint = "=".join(parts).strip().strip('"')
|
|
continue
|
|
|
|
# Azure requires the deployment name, the API key and the endpoint URL.
|
|
if include_deployment:
|
|
assert (
|
|
deployment is not None
|
|
), "Azure OpenAI deployment name not found in .env file"
|
|
|
|
assert api_key is not None, "Azure OpenAI API key not found in .env file"
|
|
assert endpoint is not None, "Azure OpenAI endpoint not found in .env file"
|
|
|
|
return deployment or "", api_key, endpoint
|