Пофиксил линтер

ai-forever · Oct 16, 2024 · fc4db18 · fc4db18
1 parent 8dbf455
commit fc4db18
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/libs/langchain_gigachat/langchain_gigachat/utils/function_calling.py b/libs/langchain_gigachat/langchain_gigachat/utils/function_calling.py
@@ -102,7 +102,7 @@ def _model_to_schema(model: Type[BaseModel]) -> dict:
 
 
 def _convert_return_schema(
-    return_model: Type[BaseModel] | dict[str, Any] | None,
+    return_model: Optional[Union[Type[BaseModel], dict[str, Any]]],
 ) -> Dict[str, Any]:
     if not return_model:
         return {}

diff --git a/libs/langchain_gigachat/langchain_gigachat/utils/pydantic_generator.py b/libs/langchain_gigachat/langchain_gigachat/utils/pydantic_generator.py
@@ -1,12 +1,12 @@
+from typing import Union
+
 from pydantic.json_schema import GenerateJsonSchema, JsonSchemaValue, core_schema
 
 
 class GigaChatJsonSchema(GenerateJsonSchema):
     def field_is_required(
         self,
-        field: core_schema.ModelField
-        | core_schema.DataclassField
-        | core_schema.TypedDictField,
+        field: Union[core_schema.ModelField, core_schema.DataclassField, core_schema.TypedDictField],
         total: bool,
     ) -> bool:
         """