Skip to content

Commit 7c4aa68

Browse files
committed
feat(chains): remove unused imports
1 parent b48b0ee commit 7c4aa68

File tree

2 files changed

+26
-44
lines changed

2 files changed

+26
-44
lines changed

src/lib/chains/imageSearchAgent.ts

Lines changed: 15 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ import {
33
RunnableMap,
44
RunnableLambda,
55
} from '@langchain/core/runnables';
6-
import { ChatPromptTemplate, PromptTemplate } from '@langchain/core/prompts';
6+
import { ChatPromptTemplate } from '@langchain/core/prompts';
77
import formatChatHistoryAsString from '../utils/formatHistory';
88
import { BaseMessage } from '@langchain/core/messages';
99
import { StringOutputParser } from '@langchain/core/output_parsers';
@@ -43,43 +43,34 @@ const createImageSearchChain = (llm: BaseChatModel) => {
4343
ChatPromptTemplate.fromMessages([
4444
['system', imageSearchChainPrompt],
4545
[
46-
"user",
47-
"<conversation>\n</conversation>\n<follow_up>\nWhat is a cat?\n</follow_up>"
48-
],
49-
[
50-
"assistant",
51-
"<query>A cat</query>"
46+
'user',
47+
'<conversation>\n</conversation>\n<follow_up>\nWhat is a cat?\n</follow_up>',
5248
],
49+
['assistant', '<query>A cat</query>'],
5350

5451
[
55-
"user",
56-
"<conversation>\n</conversation>\n<follow_up>\nWhat is a car? How does it work?\n</follow_up>"
57-
],
58-
[
59-
"assistant",
60-
"<query>Car working</query>"
61-
],
62-
[
63-
"user",
64-
"<conversation>\n</conversation>\n<follow_up>\nHow does an AC work?\n</follow_up>"
52+
'user',
53+
'<conversation>\n</conversation>\n<follow_up>\nWhat is a car? How does it work?\n</follow_up>',
6554
],
55+
['assistant', '<query>Car working</query>'],
6656
[
67-
"assistant",
68-
"<query>AC working</query>"
57+
'user',
58+
'<conversation>\n</conversation>\n<follow_up>\nHow does an AC work?\n</follow_up>',
6959
],
60+
['assistant', '<query>AC working</query>'],
7061
[
7162
'user',
72-
'<conversation>{chat_history}</conversation>\n<follow_up>\n{query}\n</follow_up>'
73-
]
63+
'<conversation>{chat_history}</conversation>\n<follow_up>\n{query}\n</follow_up>',
64+
],
7465
]),
7566
llm,
7667
strParser,
7768
RunnableLambda.from(async (input: string) => {
7869
const queryParser = new LineOutputParser({
79-
key: 'query'
80-
})
70+
key: 'query',
71+
});
8172

82-
return (await queryParser.parse(input))
73+
return await queryParser.parse(input);
8374
}),
8475
RunnableLambda.from(async (input: string) => {
8576
const res = await searchSearxng(input, {

src/lib/chains/videoSearchAgent.ts

Lines changed: 11 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ import {
33
RunnableMap,
44
RunnableLambda,
55
} from '@langchain/core/runnables';
6-
import { ChatPromptTemplate, PromptTemplate } from '@langchain/core/prompts';
6+
import { ChatPromptTemplate } from '@langchain/core/prompts';
77
import formatChatHistoryAsString from '../utils/formatHistory';
88
import { BaseMessage } from '@langchain/core/messages';
99
import { StringOutputParser } from '@langchain/core/output_parsers';
@@ -45,40 +45,31 @@ const createVideoSearchChain = (llm: BaseChatModel) => {
4545
['system', videoSearchChainPrompt],
4646
[
4747
'user',
48-
'<conversation>\n</conversation>\n<follow_up>\nHow does a car work?\n</follow_up>'
49-
],
50-
[
51-
'assistant',
52-
'<query>How does a car work?</query>'
48+
'<conversation>\n</conversation>\n<follow_up>\nHow does a car work?\n</follow_up>',
5349
],
50+
['assistant', '<query>How does a car work?</query>'],
5451
[
5552
'user',
56-
'<conversation>\n</conversation>\n<follow_up>\nWhat is the theory of relativity?\n</follow_up>'
57-
],
58-
[
59-
'assistant',
60-
'<query>Theory of relativity</query>'
53+
'<conversation>\n</conversation>\n<follow_up>\nWhat is the theory of relativity?\n</follow_up>',
6154
],
55+
['assistant', '<query>Theory of relativity</query>'],
6256
[
6357
'user',
64-
'<conversation>\n</conversation>\n<follow_up>\nHow does an AC work?\n</follow_up>'
65-
],
66-
[
67-
'assistant',
68-
'<query>AC working</query>'
58+
'<conversation>\n</conversation>\n<follow_up>\nHow does an AC work?\n</follow_up>',
6959
],
60+
['assistant', '<query>AC working</query>'],
7061
[
7162
'user',
72-
'<conversation>{chat_history}</conversation>\n<follow_up>\n{query}\n</follow_up>'
73-
]
63+
'<conversation>{chat_history}</conversation>\n<follow_up>\n{query}\n</follow_up>',
64+
],
7465
]),
7566
llm,
7667
strParser,
7768
RunnableLambda.from(async (input: string) => {
7869
const queryParser = new LineOutputParser({
79-
key: 'query'
70+
key: 'query',
8071
});
81-
return (await queryParser.parse(input));
72+
return await queryParser.parse(input);
8273
}),
8374
RunnableLambda.from(async (input: string) => {
8475
const res = await searchSearxng(input, {

0 commit comments

Comments
 (0)