Captainspa/grounded-code

1# This file is to fetch a model using the various APIs2from dotenv import load_dotenv3from os import getenv4load_dotenv()5OPENAI_API_KEY = getenv("OPENAI_API_KEY")6ANTHROPIC_API_KEY = getenv("ANTHROPIC_API_KEY")7def check_api_keys():8    assert OPENAI_API_KEY is not None and ANTHROPIC_API_KEY is not None, "API keys not found"9check_api_keys()10 11from langchain_openai import OpenAIEmbeddings12from langchain_anthropic import ChatAnthropic13from langchain.callbacks.manager import CallbackManager14from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler15 16def get_claude_sonnet():17    return ChatAnthropic(18        model_name="claude-3-sonnet-20240229",19        anthropic_api_key=ANTHROPIC_API_KEY, 20        temperature=0, 21        streaming=True,22        callback_manager=CallbackManager([StreamingStdOutCallbackHandler()])23    )24 25def get_claude_opus():26    return ChatAnthropic(27        model_name="claude-3-opus-20240229",28        anthropic_api_key=ANTHROPIC_API_KEY, 29        temperature=0, 30        streaming=True,31        callback_manager=CallbackManager([StreamingStdOutCallbackHandler()])32    )33 34def get_openai_embedder_small():35    return OpenAIEmbeddings(36        model="text-embedding-3-small",37        api_key=OPENAI_API_KEY38    )39 40def get_openai_embedder_large():41    return OpenAIEmbeddings(42        model="text-embedding-3-large",43        api_key=OPENAI_API_KEY44    )45