hotfix openai.py error

modelscope · Jun 14, 2024 · 491c1ca · 491c1ca
1 parent 1ed65eb
commit 491c1ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modelscope_agent/llm/openai.py b/modelscope_agent/llm/openai.py
@@ -159,7 +159,7 @@ def chat_with_functions(self,
 
 
 @register_llm('vllm-server')
-class Vllm(BaseChatModel):
+class Vllm(OpenAi):
 
     def _chat_stream(self,
                      messages: List[Dict],