fix file preview in file management (#1151)

### What problem does this PR solve?

fix file preview in file management

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
KevinHuSh
2024-06-14 10:33:59 +08:00
committed by GitHub
parent 64c83f300a
commit 2023fdc13e
10 changed files with 203 additions and 11 deletions

View File

@ -78,7 +78,7 @@ class GptV4(Base):
prompt = self.prompt(b64)
for i in range(len(prompt)):
for c in prompt[i]["content"]:
if "type" in c: del c["type"]
if "text" in c: c["type"] = "text"
res = self.client.chat.completions.create(
model=self.model_name,

View File

@ -15,7 +15,6 @@
#
import re
from typing import Optional
import requests
from huggingface_hub import snapshot_download
from zhipuai import ZhipuAI
@ -344,4 +343,4 @@ class InfinityEmbed(Base):
def encode_queries(self, text: str) -> tuple[np.ndarray, int]:
# Using the internal tokenizer to encode the texts and get the total
# number of tokens
return self.encode([text])
return self.encode([text])