🐛 fix(BaiduQianfanChatEndpoints.py): add type ignore comments to fix type errors

🐛 fix(BaiduQianfanLLMEndpoints.py): add type ignore comments to fix type errors
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-17 10:13:49 -03:00
commit 9b36e64c57
2 changed files with 4 additions and 4 deletions

View file

@ -78,7 +78,7 @@ class QianfanChatEndpointComponent(CustomComponent):
endpoint: Optional[str] = None,
) -> BaseLLM:
try:
output = QianfanChatEndpoint(
output = QianfanChatEndpoint( # type: ignore
model=model,
qianfan_ak=qianfan_ak,
qianfan_sk=qianfan_sk,
@ -89,4 +89,4 @@ class QianfanChatEndpointComponent(CustomComponent):
)
except Exception as e:
raise ValueError("Could not connect to Baidu Qianfan API.") from e
return output
return output # type: ignore

View file

@ -78,7 +78,7 @@ class QianfanLLMEndpointComponent(CustomComponent):
endpoint: Optional[str] = None,
) -> BaseLLM:
try:
output = QianfanLLMEndpoint(
output = QianfanLLMEndpoint( # type: ignore
model=model,
qianfan_ak=qianfan_ak,
qianfan_sk=qianfan_sk,
@ -89,4 +89,4 @@ class QianfanLLMEndpointComponent(CustomComponent):
)
except Exception as e:
raise ValueError("Could not connect to Baidu Qianfan API.") from e
return output
return output # type: ignore