diff --git a/kag/bridge/spg_server_bridge.py b/kag/bridge/spg_server_bridge.py index 7fde8f72..48d53c44 100644 --- a/kag/bridge/spg_server_bridge.py +++ b/kag/bridge/spg_server_bridge.py @@ -16,7 +16,6 @@ def init_kag_config(project_id: str, host_addr: str): - os.environ[KAGConstants.ENV_KAG_PROJECT_ID] = project_id os.environ[KAGConstants.ENV_KAG_PROJECT_HOST_ADDR] = host_addr init_env() @@ -47,3 +46,15 @@ def run_component(self, component_name, component_config, input_data): if hasattr(instance.input_types, "from_dict"): input_data = instance.input_types.from_dict(input_data) return [x.to_dict() for x in instance.invoke(input_data, write_ckpt=False)] + + def run_llm_config_check(self, llm_config): + from kag.common.llm.llm_config_checker import LLMConfigChecker + + return LLMConfigChecker().check(llm_config) + + def run_vectorizer_config_check(self, vec_config): + from kag.common.vectorize_model.vectorize_model_config_checker import ( + VectorizeModelConfigChecker, + ) + + return VectorizeModelConfigChecker().check(vec_config)