Skip to content

Commit 1016b7c

Browse files
committed
fix: Code smell fixes in utils.py
1 parent f47c375 commit 1016b7c

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

cypher_ai_helper/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def pydantic_models_to_str(models: list[Union[BaseModel, Type[BaseModel]]]) -> s
5151
return '\n\n'.join([pydantic_model_to_str(model) for model in models])
5252

5353

54-
def class_definition_in_str_for_dataclasses(model: Any):
54+
def class_definition_in_str_for_dataclasses(model: Any) -> str:
5555
"""
5656
Creates schema for dataclass
5757
@@ -109,7 +109,7 @@ def extract_params(doc_str: str) -> dict[str, str]:
109109
return params
110110

111111

112-
def func_to_json(func):
112+
def func_to_json(func) -> dict[str, Any]:
113113
"""
114114
Converts a function to a json schema for OpenAI
115115

0 commit comments

Comments
 (0)