Fix keys of Xinference deployed models, especially has the same model name with public hosted models. (#2832)

### What problem does this PR solve?

Fix keys of Xinference deployed models, especially has the same model
name with public hosted models.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

---------

Co-authored-by: 0000sir <0000sir@gmail.com>
Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
This commit is contained in:
0000sir
2024-10-16 10:21:08 +08:00
committed by GitHub
parent 51ecda0ff5
commit 4991107822
7 changed files with 22 additions and 15 deletions

View File

@ -74,6 +74,12 @@ class RAGFlow:
if res.get("code") != 0:
raise Exception(res["message"])
def get_dataset(self,name: str):
_list = self.list_datasets(name=name)
if len(_list) > 0:
return _list[0]
raise Exception("Dataset %s not found" % name)
def list_datasets(self, page: int = 1, page_size: int = 1024, orderby: str = "create_time", desc: bool = True,
id: str = None, name: str = None) -> \
List[DataSet]: