klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bug Fix] Make claude-3-opus model work (#1331)

Browse Source
This commit is contained in:
Deshraj Yadav
2024-03-28 00:56:14 -07:00
committed by GitHub
parent 85ec61335a
commit d00a2085d5
5 changed files with 10 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
embedchain/llm/anthropic.py
View File

@@ -2,6 +2,11 @@ import logging
import os
from typing import Optional
try:
from langchain_anthropic import ChatAnthropic
except ImportError:
raise ImportError("Please install the langchain-anthropic package by running `pip install langchain-anthropic`.")
from embedchain.config import BaseLlmConfig
from embedchain.helpers.json_serializable import register_deserializable
from embedchain.llm.base import BaseLlm
@@ -21,10 +26,9 @@ class AnthropicLlm(BaseLlm):
@staticmethod
def _get_answer(prompt: str, config: BaseLlmConfig) -> str:
from langchain_community.chat_models import ChatAnthropic
chat = ChatAnthropic(
anthropic_api_key=os.environ["ANTHROPIC_API_KEY"], temperature=config.temperature, model=config.model
anthropic_api_key=os.environ["ANTHROPIC_API_KEY"], temperature=config.temperature, model_name=config.model
)
if config.max_tokens and config.max_tokens != 1000: