From 8e494129400f5b468671156b425b38a5fd996286 Mon Sep 17 00:00:00 2001
From: hexists <hexists@daum.net>
Date: Sun, 27 Oct 2024 09:27:54 +0900
Subject: [PATCH] fix call get_model in ./configs/parallelqa/tools.py

---
 configs/parallelqa/tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/parallelqa/tools.py b/configs/parallelqa/tools.py
index 08e8fad..36e64b2 100644
--- a/configs/parallelqa/tools.py
+++ b/configs/parallelqa/tools.py
@@ -81,7 +81,7 @@ async def run_llm_math_chain(question, context=None):
 def generate_tools(args, model_name):
     llm_math_chain = get_model(
         model_type=args.model_type,
-        api_key=args.api_key,
+        model_name=args.model_name,
         vllm_port=args.vllm_port,
         stream=False,
         temperature=0,