Skip to content

Commit 9cce4d4

Browse files
committed
preserving cohere response citations
- this gets cohere specific response field that includes citations for the response text
1 parent ff6ab73 commit 9cce4d4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

willa/chatbot/graph_manager.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ class WillaChatbotState(TypedDict):
2323
search_query: NotRequired[str]
2424
tind_metadata: NotRequired[str]
2525
documents: NotRequired[list[Any]]
26+
citations: NotRequired[list[dict[str, Any]]]
2627
context: NotRequired[dict[str, Any]]
2728

2829

0 commit comments

Comments
 (0)