mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2026-05-17 22:13:51 +00:00
18 lines
739 B
Python
18 lines
739 B
Python
from model.api_config import api_config
|
|
global_openai_client = api_config.global_openai_client()
|
|
|
|
from model.model_config import model_registry
|
|
from model.query_manager import query_manager
|
|
def _create_query_function(model_key: str):
|
|
def query_func(messages, system_prompt=None):
|
|
return query_manager.query(model_key, messages, system_prompt)
|
|
return query_func
|
|
|
|
_generated_functions = {}
|
|
for model_key, config in model_registry.get_all_models().items():
|
|
func = _create_query_function(model_key)
|
|
func.__name__ = config.function_name
|
|
_generated_functions[config.function_name] = func
|
|
globals()[config.function_name] = func
|
|
|
|
__all__ = ['ModelQueryManager', 'query_manager'] + list(_generated_functions.keys()) |