Update provider_interface.py

This commit is contained in:
tehcoderer
2023-08-04 14:50:40 -04:00
parent b90cbc98d8
commit 4de03eafe5

View File

@@ -435,16 +435,12 @@ class ProviderInterface:
class Config(BaseConfig):
extra = Extra.allow
ReturnModel = create_model( # type: ignore
result[model_name] = create_model( # type: ignore
model_name,
__config__=Config,
**fields_dict, # type: ignore
)
# TODO: If we want to support multiple return schemas, we need
# to change this `List` to the schema type
result[model_name] = List[ReturnModel] # type: ignore
return result
def _get_provider_literal(self, available_providers: List[str]) -> type: