fix llm query in image patch

Author: ControlNet

hydra_vl4ai/execution/image_patch.py

@@ -13,7 +13,7 @@
 import tensorneko_util as N
 
 from .toolbox import forward
-from ..agent.llm import chatgpt
+from ..agent.llm import llm
 from ..agent.smb import StateMemoryBank
 from ..util.misc import get_hydra_root_folder, load_json
 from ..util.config import Config
@@ -558,10 +558,10 @@ def llm_query(query, context=None, long_answer=True, state_memory_bank=None):
     prompt_ += f'Could you help me answer the question: {query}.'
 
     if not long_answer:
-        prompt_ += f'Please provide only a few-word answer. Be very concise, no ranges, no doubt.'
+        prompt_ += 'Please provide only a few-word answer. Be very concise, no ranges, no doubt.'
     try:
-        return_answer = asyncio.run(chatgpt("gpt-3.5-turbo-1106", prompt_)) or ""
-    except:
+        return_answer = asyncio.run(llm(Config.base_config["llm_model"], prompt_)) or ""
+    except Exception:
         return_answer = 'not answer from gpt'
 
     # get global description.

hydra_vl4ai/util/config.py

@@ -2,6 +2,7 @@
 from tensorneko_util.util import Singleton
 import tensorneko_util as N
 
+
 @Singleton
 class Config:
 
@@ -22,7 +23,6 @@ def base_config_path(self, value):
             self._base_config_path = value
             self.base_config = N.read(value)
 
-
     @property
     def dqn_config_path(self):
         return self._dqn_config_path

main.py

@@ -32,7 +32,7 @@
 
 
 async def main():
-    with console.status("[bold green]Connect to HYDRA executor...") as status:
+    with console.status("[bold green]Connect to HYDRA executor..."):
         wait_until_loaded(f"http://localhost:{Config.base_config['executor_port']}")
 
     if args.dqn_config is None: