add anthropic page (#10666)

This commit is contained in:
Harrison Chase
2023-09-18 11:10:44 -07:00
committed by GitHub
parent 5072138893
commit e404fd39dd
3 changed files with 173 additions and 0 deletions

View File

@@ -15,6 +15,7 @@ from langchain.schema.messages import (
SystemMessage,
)
from langchain.schema.output import ChatGeneration, ChatGenerationChunk, ChatResult
from langchain.schema.prompt import PromptValue
def _convert_one_message_to_text(
@@ -112,6 +113,9 @@ class ChatAnthropic(BaseChatModel, _AnthropicCommon):
prompt_params["ai_prompt"] = self.AI_PROMPT
return convert_messages_to_prompt_anthropic(messages=messages, **prompt_params)
def convert_prompt(self, prompt: PromptValue) -> str:
return self._convert_messages_to_prompt(prompt.to_messages())
def _stream(
self,
messages: List[BaseMessage],

View File

@@ -10,6 +10,7 @@ from langchain.llms.base import LLM
from langchain.pydantic_v1 import Field, root_validator
from langchain.schema.language_model import BaseLanguageModel
from langchain.schema.output import GenerationChunk
from langchain.schema.prompt import PromptValue
from langchain.utils import (
check_package_version,
get_from_dict_or_env,
@@ -234,6 +235,9 @@ class Anthropic(LLM, _AnthropicCommon):
)
return response.completion
def convert_prompt(self, prompt: PromptValue) -> str:
return self._wrap_prompt(prompt.to_string())
async def _acall(
self,
prompt: str,