Skip to content

Commit

Permalink
adds anthropic model to instantiation method
Browse files Browse the repository at this point in the history
  • Loading branch information
lauraschauer committed Jul 10, 2024
1 parent dcc1847 commit a3a1c46
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions prospector/llm/instantiation.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,13 @@

import requests
from dotenv import load_dotenv
from langchain_anthropic import ChatAnthropic
from langchain_core.language_models.llms import LLM
from langchain_google_vertexai import ChatVertexAI
from langchain_mistralai import ChatMistralAI
from langchain_openai import ChatOpenAI

from llm.models.anthropic import Anthropic
from llm.models.gemini import Gemini
from llm.models.mistral import Mistral
from llm.models.openai import OpenAI
Expand All @@ -26,6 +28,7 @@
# "gpt-4o": OpenAI, # currently TBD
"gemini-1.0-pro": Gemini,
"mistral-large": Mistral,
"claude-3-opus": Anthropic,
}


Expand All @@ -37,6 +40,7 @@
"gpt-3.5-turbo": (ChatOpenAI, "OPENAI_API_KEY"),
"gemini-pro": (ChatVertexAI, "GOOGLE_API_KEY"),
"mistral-large-latest": (ChatMistralAI, "MISTRAL_API_KEY"),
"claude-3-opus-20240229": (ChatAnthropic, "ANTRHOPIC_API_KEY"),
}


Expand Down

0 comments on commit a3a1c46

Please sign in to comment.