diff --git a/.vscode/settings.json b/.vscode/settings.json index 7b48ec4c..6d49d14c 100644 --- a/.vscode/settings.json +++ b/.vscode/settings.json @@ -10,6 +10,7 @@ "Izakaya", "Jajarkot", "jsonv", + "Juárez", "Karnali", "Kita", "Kurunegala", diff --git a/src/utils/langchain/chains/deep/prompt.ts b/src/utils/langchain/chains/deep/prompt.ts index b333d5fb..7dd4f129 100644 --- a/src/utils/langchain/chains/deep/prompt.ts +++ b/src/utils/langchain/chains/deep/prompt.ts @@ -13,25 +13,37 @@ export const tridentDeepExampleList: Array<{ { input: "Area: Sudan", output: `[out:json][timeout:30000]; -relation["boundary"="administrative"]["name"="Sudan"]; +relation["boundary"="administrative"]["admin_level"=2]["name"="Sudan"]; out geom;`, }, { input: "Area: Lebanon", output: `[out:json][timeout:30000]; -relation["boundary"="administrative"]["name"="Lebanon"]; +relation["boundary"="administrative"]["admin_level"=2]["name"="Lebanon"]; +out geom;`, + }, + { + input: "Area: Kosovo", + output: `[out:json][timeout:30000]; +relation["boundary"="administrative"]["admin_level"=2]["name"="Kosovo"]; +out geom;`, + }, + { + input: "Area: Nepal", + output: `[out:json][timeout:30000]; +relation["boundary"="administrative"]["admin_level"=2]["name"="Nepal"]; out geom;`, }, { input: "Area: New York City", output: `[out:json][timeout:30000]; -relation["boundary"="administrative"]["name"="City of New York"]; +relation["boundary"="administrative"]["admin_level"=5]["name"="City of New York"]; out geom;`, }, { input: "Area: Tokyo", output: `[out:json][timeout:30000]; -relation["boundary"="administrative"]["name"="Tokyo"]; +relation["boundary"="administrative"]["admin_level"=4]["name"="Tokyo"]; out geom;`, }, { @@ -95,7 +107,16 @@ out geom;`, area["name:en"="Nepal"]->.outer; area["name:en"="Karnali Province"]->.inner; ( - relation["boundary"="administrative"]["name:en"="Western Rukum District"](area.inner)(area.outer); + relation["boundary"="administrative"]["name"="Western Rukum District"](area.inner)(area.outer); +); +out geom;`, + }, + { + input: "Nepal, UNICEF facilities", + output: `[out:json][timeout:30000]; +area["name"="Nepal"]->.searchArea; +( + nwr["name"~"UNICEF"](area.searchArea); ); out geom;`, }, @@ -371,7 +392,7 @@ export const loadTridentDeepPrompt = async (embeddings: Embeddings) => { const memoryVectorStore = new MemoryVectorStore(embeddings); const exampleSelector = new SemanticSimilarityExampleSelector({ vectorStore: memoryVectorStore, - k: 3, + k: 5, inputKeys: ["input"], }); const examplePrompt = PromptTemplate.fromTemplate( diff --git a/src/utils/langchain/chains/inner/prompt.ts b/src/utils/langchain/chains/inner/prompt.ts index c5634d15..0ed36821 100644 --- a/src/utils/langchain/chains/inner/prompt.ts +++ b/src/utils/langchain/chains/inner/prompt.ts @@ -105,6 +105,15 @@ ColorForConcern: Embassies, lightblue AreaWithConcern: Lebanon, Military facilities EmojiForConcern: Military facilities, 🪖 ColorForConcern: Military facilities, yellow`, + }, + { + input: "コソボの大使館を表示して", + output: `ConfirmHelpful: 地図の作成が完了しました。他にご要望はありますか?私たちは皆さんのお役に立つことができましたでしょうか? +TitleOfMap: コソボの大使館 +Area: Kosovo +AreaWithConcern: Kosovo, Embassies +EmojiForConcern: Embassies, 🏢 +ColorForConcern: Embassies, lightblue`, }, { input: `ネパールのユニセフの施設の地図を表示します。`,