github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-10 15:33:11 +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

12
libs/community/langchain_community/chat_models/azureml_endpoint.py
View File

@@ -20,7 +20,7 @@ from langchain_community.llms.azureml_endpoint import (
class LlamaContentFormatter(ContentFormatterBase):
def __init__(self):
def __init__(self): # type: ignore[no-untyped-def]
raise TypeError(
"`LlamaContentFormatter` is deprecated for chat models. Use "
"`LlamaChatContentFormatter` instead."
@@ -72,7 +72,7 @@ class LlamaChatContentFormatter(ContentFormatterBase):
def supported_api_types(self) -> List[AzureMLEndpointApiType]:
return [AzureMLEndpointApiType.realtime, AzureMLEndpointApiType.serverless]
def format_request_payload(
def format_request_payload( # type: ignore[override]
self,
messages: List[BaseMessage],
model_kwargs: Dict,
@@ -98,9 +98,9 @@ class LlamaChatContentFormatter(ContentFormatterBase):
raise ValueError(
f"`api_type` {api_type} is not supported by this formatter"
)
return str.encode(request_payload)
return str.encode(request_payload) # type: ignore[return-value]
def format_response_payload(
def format_response_payload( # type: ignore[override]
self, output: bytes, api_type: AzureMLEndpointApiType
) -> ChatGeneration:
"""Formats response"""
@@ -108,7 +108,7 @@ class LlamaChatContentFormatter(ContentFormatterBase):
try:
choice = json.loads(output)["output"]
except (KeyError, IndexError, TypeError) as e:
raise ValueError(self.format_error_msg.format(api_type=api_type)) from e
raise ValueError(self.format_error_msg.format(api_type=api_type)) from e # type: ignore[union-attr]
return ChatGeneration(
message=BaseMessage(
content=choice.strip(),
@@ -125,7 +125,7 @@ class LlamaChatContentFormatter(ContentFormatterBase):
"model. Expected `dict` but `{type(choice)}` was received."
)
except (KeyError, IndexError, TypeError) as e:
raise ValueError(self.format_error_msg.format(api_type=api_type)) from e
raise ValueError(self.format_error_msg.format(api_type=api_type)) from e # type: ignore[union-attr]
return ChatGeneration(
message=BaseMessage(
content=choice["message"]["content"].strip(),