mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-08 20:42:30 +08:00
add dockerfile for cuda envirement. Refine table search strategy, (#123)
This commit is contained in:
@ -14,11 +14,8 @@
|
||||
# limitations under the License.
|
||||
#
|
||||
from abc import ABC
|
||||
from copy import deepcopy
|
||||
|
||||
from openai import OpenAI
|
||||
import openai
|
||||
|
||||
from rag.nlp import is_english
|
||||
from rag.utils import num_tokens_from_string
|
||||
|
||||
@ -52,6 +49,12 @@ class GptTurbo(Base):
|
||||
return "**ERROR**: "+str(e), 0
|
||||
|
||||
|
||||
class MoonshotChat(GptTurbo):
|
||||
def __init__(self, key, model_name="moonshot-v1-8k"):
|
||||
self.client = OpenAI(api_key=key, base_url="https://api.moonshot.cn/v1",)
|
||||
self.model_name = model_name
|
||||
|
||||
|
||||
from dashscope import Generation
|
||||
class QWenChat(Base):
|
||||
def __init__(self, key, model_name=Generation.Models.qwen_turbo):
|
||||
|
||||
Reference in New Issue
Block a user