Fix

chain-ml · Jan 15, 2025 · 471a32e · 471a32e
1 parent ff94008
commit 471a32e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/council/llm/llm_function/llm_function_with_prompt.py b/council/llm/llm_function/llm_function_with_prompt.py
@@ -4,7 +4,7 @@
 from typing import Any, Iterable, Mapping, Optional, Union
 
 from council.llm.base import LLMBase, LLMCacheControlData, LLMMessage, get_llm_from_config
-from council.prompt import LLMPromptConfigObjectBase
+from council.prompt import LLMPromptConfigObject, LLMPromptConfigObjectBase
 
 from .llm_function import LLMFunction, LLMFunctionResponse, LLMResponseParser, T_Response
 from .llm_middleware import LLMMiddlewareChain
@@ -101,7 +101,8 @@ def from_configs(
         """
 
         llm = get_llm_from_config(os.path.join(path_prefix, llm_path))
-        prompt_config = LLMPromptConfigObjectBase.from_yaml(os.path.join(path_prefix, prompt_config_path))
+        # TODO: hard-coded for not structured prompt config
+        prompt_config = LLMPromptConfigObject.from_yaml(os.path.join(path_prefix, prompt_config_path))
         return LLMFunctionWithPrompt(
             llm, response_parser, prompt_config, max_retries, system_prompt_params, system_prompt_caching
         )