github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-07 14:03:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: upgrade to recent version of mypy (#21616)

Browse Source 
This PR upgrades community to a recent version of mypy. It inserts type:
ignore on all existing failures.
This commit is contained in:
Eugene Yurtsev
2024-05-13 14:55:07 -04:00
committed by GitHub
parent b923951062
commit 25fbe356b4
243 changed files with 718 additions and 710 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/community/langchain_community/llms/textgen.py
View File

@@ -330,13 +330,13 @@ class TextGen(LLM):
result = websocket_client.recv()
result = json.loads(result)
if result["event"] == "text_stream":
if result["event"] == "text_stream": # type: ignore[call-overload, index]
chunk = GenerationChunk(
text=result["text"],
text=result["text"], # type: ignore[call-overload, index]
generation_info=None,
)
yield chunk
elif result["event"] == "stream_end":
elif result["event"] == "stream_end": # type: ignore[call-overload, index]
websocket_client.close()
return
@@ -403,13 +403,13 @@ class TextGen(LLM):
result = websocket_client.recv()
result = json.loads(result)
if result["event"] == "text_stream":
if result["event"] == "text_stream": # type: ignore[call-overload, index]
chunk = GenerationChunk(
text=result["text"],
text=result["text"], # type: ignore[call-overload, index]
generation_info=None,
)
yield chunk
elif result["event"] == "stream_end":
elif result["event"] == "stream_end": # type: ignore[call-overload, index]
websocket_client.close()
return