mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-19 21:33:31 -04:00
Leonid Ganeline
1c81883d42
This PR targets the `API Reference` documentation.
- Several classes and functions missed `docstrings`. These docstrings
were created.
- In several places this
```
except ImportError:
raise ValueError(
```
was replaced to
```
except ImportError:
raise ImportError(
```
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import Callable, List, Tuple
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from langchain.base_language import BaseLanguageModel
|
|
from langchain.chat_models.base import BaseChatModel
|
|
from langchain.llms.base import BaseLLM
|
|
from langchain.prompts.base import BasePromptTemplate
|
|
|
|
|
|
class BasePromptSelector(BaseModel, ABC):
|
|
@abstractmethod
|
|
def get_prompt(self, llm: BaseLanguageModel) -> BasePromptTemplate:
|
|
"""Get default prompt for a language model."""
|
|
|
|
|
|
class ConditionalPromptSelector(BasePromptSelector):
|
|
"""Prompt collection that goes through conditionals."""
|
|
|
|
default_prompt: BasePromptTemplate
|
|
conditionals: List[
|
|
Tuple[Callable[[BaseLanguageModel], bool], BasePromptTemplate]
|
|
] = Field(default_factory=list)
|
|
|
|
def get_prompt(self, llm: BaseLanguageModel) -> BasePromptTemplate:
|
|
for condition, prompt in self.conditionals:
|
|
if condition(llm):
|
|
return prompt
|
|
return self.default_prompt
|
|
|
|
|
|
def is_llm(llm: BaseLanguageModel) -> bool:
|
|
"""Check if the language model is a LLM.
|
|
|
|
Args:
|
|
llm: Language model to check.
|
|
|
|
Returns:
|
|
True if the language model is a BaseLLM model, False otherwise.
|
|
"""
|
|
return isinstance(llm, BaseLLM)
|
|
|
|
|
|
def is_chat_model(llm: BaseLanguageModel) -> bool:
|
|
"""Check if the language model is a chat model.
|
|
|
|
Args:
|
|
llm: Language model to check.
|
|
|
|
Returns:
|
|
True if the language model is a BaseChatModel model, False otherwise.
|
|
"""
|
|
return isinstance(llm, BaseChatModel)
|