sk/Perplexica
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(agents): use ollama models

Browse source
This commit is contained in:
ItzCrazyKns 2024-04-17 10:22:20 +05:30
parent 89340fcfff
commit 811822c03b
No known key found for this signature in database
GPG key ID: 8162927C7CCE3065
8 changed files with 65 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/agents/imageSearchAgent.ts
View file

@ -4,15 +4,15 @@ import {
RunnableLambda,
} from '@langchain/core/runnables';
import { PromptTemplate } from '@langchain/core/prompts';
import { OpenAI } from '@langchain/openai';
import { Ollama } from '@langchain/community/llms/ollama';
import formatChatHistoryAsString from '../utils/formatHistory';
import { BaseMessage } from '@langchain/core/messages';
import { StringOutputParser } from '@langchain/core/output_parsers';
import { searchSearxng } from '../core/searxng';
const llm = new OpenAI({
const llm = new Ollama({
temperature: 0,
modelName: process.env.MODEL_NAME,
model: process.env.MODEL_NAME,
});
const imageSearchChainPrompt = `