fix: add completion_request.documents to chat_history

0xPlaygrounds · Sep 23, 2024 · 6dd9cd1 · 6dd9cd1
1 parent c708e1f
commit 6dd9cd1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rig-core/src/providers/anthropic/completion.rs b/rig-core/src/providers/anthropic/completion.rs
@@ -163,6 +163,10 @@ impl completion::CompletionModel for CompletionModel {
                 .chat_history
                 .into_iter()
                 .map(Message::from)
+                .chain(completion_request.documents.into_iter().map(|doc| Message {
+                    role: "system".to_owned(),
+                    content: serde_json::to_string(&doc).expect("Document should serialize"),
+                }))
                 .chain(iter::once(Message {
                     role: "user".to_owned(),
                     content: completion_request.prompt,