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

View File

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

View File

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