### What problem does this PR solve? #1588 ### Type of change - [x] Refactoringtags/v0.12.0
| class XinferenceCV(Base): | class XinferenceCV(Base): | ||||
| def __init__(self, key, model_name="", lang="Chinese", base_url=""): | def __init__(self, key, model_name="", lang="Chinese", base_url=""): | ||||
| if base_url.split("/")[-1] != "v1": | |||||
| base_url = os.path.join(base_url, "v1") | |||||
| self.client = OpenAI(api_key="xxx", base_url=base_url) | self.client = OpenAI(api_key="xxx", base_url=base_url) | ||||
| self.model_name = model_name | self.model_name = model_name | ||||
| self.lang = lang | self.lang = lang |
| class XinferenceEmbed(Base): | class XinferenceEmbed(Base): | ||||
| def __init__(self, key, model_name="", base_url=""): | def __init__(self, key, model_name="", base_url=""): | ||||
| if base_url.split("/")[-1] != "v1": | |||||
| base_url = os.path.join(base_url, "v1") | |||||
| self.client = OpenAI(api_key="xxx", base_url=base_url) | self.client = OpenAI(api_key="xxx", base_url=base_url) | ||||
| self.model_name = model_name | self.model_name = model_name | ||||
| class XInferenceRerank(Base): | class XInferenceRerank(Base): | ||||
| def __init__(self, key="xxxxxxx", model_name="", base_url=""): | def __init__(self, key="xxxxxxx", model_name="", base_url=""): | ||||
| if base_url.split("/")[-1] != "v1": | |||||
| base_url = os.path.join(base_url, "v1") | |||||
| self.model_name = model_name | self.model_name = model_name | ||||
| self.base_url = base_url | self.base_url = base_url | ||||
| self.headers = { | self.headers = { |
| class XinferenceSeq2txt(Base): | class XinferenceSeq2txt(Base): | ||||
| def __init__(self, key, model_name="", base_url=""): | def __init__(self, key, model_name="", base_url=""): | ||||
| if base_url.split("/")[-1] != "v1": | |||||
| base_url = os.path.join(base_url, "v1") | |||||
| self.client = OpenAI(api_key="xxx", base_url=base_url) | self.client = OpenAI(api_key="xxx", base_url=base_url) | ||||
| self.model_name = model_name | self.model_name = model_name | ||||