diff --git a/community_usecase/Smart_city_research/main.py b/community_usecase/Smart_city_research/main.py index 954bc3e..7894729 100644 --- a/community_usecase/Smart_city_research/main.py +++ b/community_usecase/Smart_city_research/main.py @@ -67,6 +67,11 @@ def construct_society(question: str) -> RolePlaying: model_type=ModelType.MISTRAL_LARGE, model_config_dict=MistralConfig(temperature=0.0).as_dict(), ), + "document": ModelFactory.create( + model_platform=ModelPlatformType.MISTRAL, + model_type=ModelType.MISTRAL_LARGE, + model_config_dict=MistralConfig(temperature=0.0).as_dict(), + ), "browsing": ModelFactory.create( model_platform=ModelPlatformType.MISTRAL, model_type=ModelType.MISTRAL_LARGE,