github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-13 21:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

infra: add -p to mkdir in lint steps (#17013)

Browse Source 
Previously, if this did not find a mypy cache then it wouldnt run

this makes it always run

adding mypy ignore comments with existing uncaught issues to unblock other prs

---------

Co-authored-by: Erick Friis <erick@langchain.dev>
Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
This commit is contained in:
Harrison Chase
2024-02-05 11:22:06 -08:00
committed by GitHub
parent db6af21395
commit 4eda647fdd
103 changed files with 378 additions and 369 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
libs/community/langchain_community/chat_models/zhipuai.py
View File

@@ -161,7 +161,7 @@ class ChatZhipuAI(BaseChatModel):
return attributes
def __init__(self, *args, **kwargs):
def __init__(self, *args, **kwargs): # type: ignore[no-untyped-def]
super().__init__(*args, **kwargs)
try:
import zhipuai
@@ -174,7 +174,7 @@ class ChatZhipuAI(BaseChatModel):
"Please install it via 'pip install zhipuai'"
)
def invoke(self, prompt):
def invoke(self, prompt): # type: ignore[no-untyped-def]
if self.model == "chatglm_turbo":
return self.zhipuai.model_api.invoke(
model=self.model,
@@ -195,7 +195,7 @@ class ChatZhipuAI(BaseChatModel):
)
return None
def sse_invoke(self, prompt):
def sse_invoke(self, prompt): # type: ignore[no-untyped-def]
if self.model == "chatglm_turbo":
return self.zhipuai.model_api.sse_invoke(
model=self.model,
@@ -218,7 +218,7 @@ class ChatZhipuAI(BaseChatModel):
)
return None
async def async_invoke(self, prompt):
async def async_invoke(self, prompt): # type: ignore[no-untyped-def]
loop = asyncio.get_running_loop()
partial_func = partial(
self.zhipuai.model_api.async_invoke, model=self.model, prompt=prompt
@@ -229,7 +229,7 @@ class ChatZhipuAI(BaseChatModel):
)
return response
async def async_invoke_result(self, task_id):
async def async_invoke_result(self, task_id): # type: ignore[no-untyped-def]
loop = asyncio.get_running_loop()
response = await loop.run_in_executor(
None,
@@ -270,11 +270,14 @@ class ChatZhipuAI(BaseChatModel):
else:
stream_iter = self._stream(
prompt=prompt, stop=stop, run_manager=run_manager, **kwargs
prompt=prompt, # type: ignore[arg-type]
stop=stop,
run_manager=run_manager,
**kwargs,
)
return generate_from_stream(stream_iter)
async def _agenerate(
async def _agenerate( # type: ignore[override]
self,
messages: List[BaseMessage],
stop: Optional[List[str]] = None,
@@ -307,7 +310,7 @@ class ChatZhipuAI(BaseChatModel):
generations=[ChatGeneration(message=AIMessage(content=content))]
)
def _stream(
def _stream( # type: ignore[override]
self,
prompt: List[Dict[str, str]],
stop: Optional[List[str]] = None,