github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-01-24 05:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typing for LLMMathChain (#2183)

Browse Source 
Fix typing in LLMMathChain to allow chat models (#1834). Might have been
forgotten in related PR #1807.
This commit is contained in:
Ben Heckmann
2023-03-30 16:52:58 +02:00
committed by GitHub
parent 3207a74829
commit fd1fcb5a7d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/chains/llm_math/base.py
View File

@@ -6,9 +6,9 @@ from pydantic import BaseModel, Extra
from langchain.chains.base import Chain
from langchain.chains.llm import LLMChain
from langchain.chains.llm_math.prompt import PROMPT
from langchain.llms.base import BaseLLM
from langchain.prompts.base import BasePromptTemplate
from langchain.python import PythonREPL
from langchain.schema import BaseLanguageModel
class LLMMathChain(Chain, BaseModel):
@@ -21,7 +21,7 @@ class LLMMathChain(Chain, BaseModel):
llm_math = LLMMathChain(llm=OpenAI())
"""
llm: BaseLLM
llm: BaseLanguageModel
"""LLM wrapper to use."""
prompt: BasePromptTemplate = PROMPT
"""Prompt to use to translate to python if neccessary."""