pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

add base url for OpenAI (#166)

Browse Source
This commit is contained in:
KevinHuSh
2024-03-28 19:15:16 +08:00
committed by GitHub
parent be2b904daf
commit 38e5737067
8 changed files with 39 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
rag/llm/cv_model.py
View File

@ -67,8 +67,9 @@ class Base(ABC):
class GptV4(Base):
def __init__(self, key, model_name="gpt-4-vision-preview", lang="Chinese"):
self.client = OpenAI(api_key=key)
def __init__(self, key, model_name="gpt-4-vision-preview", lang="Chinese", base_url="https://api.openai.com/v1"):
if not base_url: base_url="https://api.openai.com/v1"
self.client = OpenAI(api_key=key, base_url=base_url)
self.model_name = model_name
self.lang = lang
@ -84,7 +85,7 @@ class GptV4(Base):
class QWenCV(Base):
def __init__(self, key, model_name="qwen-vl-chat-v1", lang="Chinese"):
def __init__(self, key, model_name="qwen-vl-chat-v1", lang="Chinese", **kwargs):
import dashscope
dashscope.api_key = key
self.model_name = model_name
@ -123,7 +124,7 @@ class QWenCV(Base):
class Zhipu4V(Base):
def __init__(self, key, model_name="glm-4v", lang="Chinese"):
def __init__(self, key, model_name="glm-4v", lang="Chinese", **kwargs):
self.client = ZhipuAI(api_key=key)
self.model_name = model_name
self.lang = lang
@ -140,7 +141,7 @@ class Zhipu4V(Base):
class LocalCV(Base):
def __init__(self, key, model_name="glm-4v", lang="Chinese"):
def __init__(self, key, model_name="glm-4v", lang="Chinese", **kwargs):
pass
def describe(self, image, max_tokens=1024):