fix lint

langchain-ai · Dec 26, 2024 · 95c9fbe · 95c9fbe
1 parent d224988
commit 95c9fbe
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 159 deletions.
diff --git a/libs/community/langchain_community/llms/modelscope_endpoint.py b/libs/community/langchain_community/llms/modelscope_endpoint.py
@@ -105,7 +105,7 @@ async def astream(self, request: Any) -> AsyncIterator[str]:
 class ModelScopeCommon(BaseModel):
     """Common parameters for Modelscope LLMs."""
 
-    client: Any
+    client: Any = Field(default=None)
     base_url: str = MODELSCOPE_SERVICE_URL_BASE
     modelscope_sdk_token: Optional[SecretStr] = Field(default=None, alias="api_key")
     model_name: str = Field(default="Qwen/Qwen2.5-Coder-32B-Instruct", alias="model")

diff --git a/libs/community/langchain_community/llms/modelscope_pipeline.py b/libs/community/langchain_community/llms/modelscope_pipeline.py
diff --git a/libs/community/tests/integration_tests/chat_models/test_modelscope_chat_endpoint.py b/libs/community/tests/integration_tests/chat_models/test_modelscope_chat_endpoint.py
@@ -6,15 +6,15 @@
 
 
 def test_modelscope_chat_call() -> None:
-    chat = ModelScopeChatEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")
+    chat = ModelScopeChatEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")  # type: ignore
     response = chat.invoke([HumanMessage(content="Say foo:")])
     assert isinstance(response, BaseMessage)
     assert isinstance(response.content, str)
 
 
 def test_modelscope_chat_multiple_history() -> None:
     """Tests multiple history works."""
-    chat = ModelScopeChatEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")
+    chat = ModelScopeChatEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")  # type: ignore
 
     response = chat.invoke(
         [
@@ -29,7 +29,7 @@ def test_modelscope_chat_multiple_history() -> None:
 
 def test_modelscope_chat_stream() -> None:
     """Test that stream works."""
-    chat = ModelScopeChatEndpoint(
+    chat = ModelScopeChatEndpoint(  # type: ignore
         model="Qwen/Qwen2.5-Coder-32B-Instruct",
         streaming=True,
     )

diff --git a/libs/community/tests/integration_tests/llms/test_modelscope_endpoint.py b/libs/community/tests/integration_tests/llms/test_modelscope_endpoint.py
@@ -7,14 +7,14 @@
 
 def test_modelscope_call() -> None:
     """Test valid call to Modelscope."""
-    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")
+    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")  # type: ignore
     output = llm.invoke("Say foo:")
     assert isinstance(output, str)
 
 
 def test_modelscope_streaming() -> None:
     """Test streaming call to Modelscope."""
-    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")
+    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")  # type: ignore
     generator = llm.stream("write a quick sort in python")
     stream_results_string = ""
     assert isinstance(generator, Iterator)
@@ -26,13 +26,13 @@ def test_modelscope_streaming() -> None:
 
 
 async def test_modelscope_call_async() -> None:
-    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")
+    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")  # type: ignore
     output = await llm.ainvoke("write a quick sort in python")
     assert isinstance(output, str)
 
 
 async def test_modelscope_streaming_async() -> None:
-    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")
+    llm = ModelScopeEndpoint(model="Qwen/Qwen2.5-Coder-32B-Instruct")  # type: ignore
     generator = llm.astream("write a quick sort in python")
     stream_results_string = ""
     assert isinstance(generator, AsyncIterator)

diff --git a/libs/community/tests/integration_tests/llms/test_modelscope_pipeline.py b/libs/community/tests/integration_tests/llms/test_modelscope_pipeline.py