Spaces:
Paused
Paused
File size: 757 Bytes
566d4e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import os
from langchain import OpenAI
from langchain.chat_models import ChatOpenAI
OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
def get_default_cloud_chat_llm():
"""
Returns a default LLM instance with the OpenAI API key set in the environment.
Returns:
OpenAI: A new OpenAI instance.
"""
llm = ChatOpenAI(model="gpt-3.5-turbo", openai_api_key=OPENAI_API_KEY, temperature=0)
return llm
def get_default_cloud_completion_llm():
"""
Returns a default LLM instance with the OpenAI API key set in the environment.
Returns:
OpenAI: A new OpenAI instance.
"""
llm = OpenAI(openai_api_key=OPENAI_API_KEY)
return llm
def get_default_local_llm():
"""
Coming soon!
"""
pass |