remi/Perplexica
1
0
Fork 0
mirror of https://github.com/ItzCrazyKns/Perplexica.git synced 2025-12-03 10:18:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(app): migrate image search chain

Browse Source
This commit is contained in:
ItzCrazyKns
2025-11-13 11:51:13 +05:30
parent 41fe009847
commit 33b736e1e8
3 changed files with 72 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/app/api/images/route.ts
View File

@@ -1,4 +1,4 @@
import handleImageSearch from '@/lib/agents/media/image';
import searchImages from '@/lib/agents/media/image';
import ModelRegistry from '@/lib/models/registry';
import { ModelWithProvider } from '@/lib/models/types';
import { AIMessage, BaseMessage, HumanMessage } from '@langchain/core/messages';
@@ -13,6 +13,13 @@ export const POST = async (req: Request) => {
try {
const body: ImageSearchBody = await req.json();
const registry = new ModelRegistry();
const llm = await registry.loadChatModel(
body.chatModel.providerId,
body.chatModel.key,
);
const chatHistory = body.chatHistory
.map((msg: any) => {
if (msg.role === 'user') {
@@ -23,16 +30,9 @@ export const POST = async (req: Request) => {
})
.filter((msg) => msg !== undefined) as BaseMessage[];
const registry = new ModelRegistry();
const llm = await registry.loadChatModel(
body.chatModel.providerId,
body.chatModel.key,
);
const images = await handleImageSearch(
const images = await searchImages(
{
chat_history: chatHistory,
chatHistory: chatHistory,
query: body.query,
},
llm,