Skip to content
Snippets Groups Projects
Commit cf1e7217 authored by Bogdan Buduroiu's avatar Bogdan Buduroiu
Browse files

Formatting

parent 441ddb93
No related branches found
No related tags found
No related merge requests found
import json import json
from typing import Any, Optional, List from typing import Any, List, Optional
from pydantic import BaseModel from pydantic import BaseModel
......
...@@ -17,7 +17,9 @@ def get_schema(item: Union[BaseModel, Callable]) -> Dict[str, Any]: ...@@ -17,7 +17,9 @@ def get_schema(item: Union[BaseModel, Callable]) -> Dict[str, Any]:
if default_value: if default_value:
default_repr = repr(default_value) default_repr = repr(default_value)
signature_part = f"{field_name}: {field_model.__name__} = {default_repr}" signature_part = (
f"{field_name}: {field_model.__name__} = {default_repr}"
)
else: else:
signature_part = f"{field_name}: {field_model.__name__}" signature_part = f"{field_name}: {field_model.__name__}"
...@@ -39,7 +41,9 @@ def get_schema(item: Union[BaseModel, Callable]) -> Dict[str, Any]: ...@@ -39,7 +41,9 @@ def get_schema(item: Union[BaseModel, Callable]) -> Dict[str, Any]:
# TODO: Add route layer object to the input, solve circular import issue # TODO: Add route layer object to the input, solve circular import issue
async def route_and_execute(query: str, llm: BaseLLM, functions: List[Callable], layer) -> Any: async def route_and_execute(
query: str, llm: BaseLLM, functions: List[Callable], layer
) -> Any:
route_choice: RouteChoice = layer(query) route_choice: RouteChoice = layer(query)
for function in functions: for function in functions:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment