Skip to content

Commit fa8d39f

Browse files
Merge pull request #12 from cpbotha/cpb-add-citations
2 parents 9e570b2 + 0f52d31 commit fa8d39f

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

src/index.ts

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,10 +232,19 @@ class PerplexityServer {
232232
messages: [{ role: "user", content: prompt }],
233233
});
234234

235+
// response.data can have a string[] .citations
236+
// these are referred to in the return text as numbered citations e.g. [1]
237+
const sourcesText = response.data.citations
238+
? `\n## Sources\nPlease keep the numbered citations inline.\n
239+
${response.data.citations
240+
.map((c: string, i: number) => `${i + 1}: ${c}`)
241+
.join("\n")}`
242+
: "";
243+
235244
return {
236245
content: [{
237246
type: "text",
238-
text: response.data.choices[0].message.content
247+
text: response.data.choices[0].message.content + sourcesText,
239248
}]
240249
};
241250
} catch (error) {

0 commit comments

Comments
 (0)