fix: image model does not work when retrieval tool is enabled (#2538)

This commit is contained in:
Louis 2024-03-29 16:07:49 +07:00 committed by GitHub
parent 132487acec
commit 228a363914
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 6 additions and 4 deletions

View File

@ -18,7 +18,7 @@ export class RetrievalTool extends InferenceTool {
tool?: AssistantTool
): Promise<MessageRequest> {
if (!data.model || !data.messages) {
return Promise.resolve(this.normalize(data))
return Promise.resolve(data)
}
const latestMessage = data.messages[data.messages.length - 1]
@ -38,6 +38,8 @@ export class RetrievalTool extends InferenceTool {
docFile,
data.model?.engine
)
} else {
return Promise.resolve(data)
}
} else if (
// Check whether we need to ingest document or not
@ -48,7 +50,7 @@ export class RetrievalTool extends InferenceTool {
) {
// No document ingested, reroute the result to inference engine
return Promise.resolve(this.normalize(data))
return Promise.resolve(data)
}
// 2. Load agent on thread changed
if (this.retrievalThreadId !== data.threadId) {

View File

@ -19,7 +19,7 @@ const AutoLink: React.FC<Props> = ({ text }) => {
key={url}
target="blank"
href={url.startsWith('http') ? url : `http://${url}`}
className="text-blue-400"
className="text-primary dark:text-blue-400"
>
{url}
</a>

View File

@ -86,7 +86,7 @@ const ErrorMessage = ({ message }: { message: ThreadMessage }) => {
Model is currently unavailable. Please switch to a different
model or install the{' '}
<button
className="font-medium text-blue-500"
className="font-medium text-primary dark:text-blue-400"
onClick={() => setMainState(MainViewState.Settings)}
>
{loadModelError.split('::')[1] ?? ''}