github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-13 16:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replicate fix linting

Browse Source
This commit is contained in:
Bagatur 2023-09-12 15:46:36 -07:00
parent 21fbbe83a7
commit 7ecee7821a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/llms/replicate.py
View File

@ -1,5 +1,5 @@
import logging
from typing import Any, Dict, List, Mapping, Optional
from typing import Any, Dict, List, Optional
from langchain.callbacks.manager import CallbackManagerForLLMRun
from langchain.llms.base import LLM
@ -82,7 +82,7 @@ class Replicate(LLM):
return values
@property
def _identifying_params(self) -> Mapping[str, Any]:
def _identifying_params(self) -> Dict[str, Any]:
"""Get the identifying parameters."""
return {
"model": self.model,
@ -126,7 +126,7 @@ class Replicate(LLM):
self.prompt_key = input_properties[0][0]
inputs = {self.prompt_key: prompt, **self.input}
inputs: Dict = {self.prompt_key: prompt, **self.input}
prediction = replicate_python.predictions.create(
version=version, input={**inputs, **kwargs}