Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-18 09:25:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Handle when documents are not provided in the Cohere response (#16144)

Browse Source 
- **Description:** This handles the cohere response when documents
aren't included in the response
  - **Issue:** N/A
  - **Dependencies:** N/A
  - **Twitter handle:** N/A
This commit is contained in:
BeatrixCohere 2024-01-17 17:11:00 +00:00 committed by GitHub
parent d91126fc64
commit b0c3e3db2b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/retrievers/cohere_rag_retriever.py
View File

@ -17,10 +17,14 @@ if TYPE_CHECKING:
def _get_docs(response: Any) -> List[Document]: def _get_docs(response: Any) -> List[Document]:
docs = [ docs = (
[]
if "documents" not in response.generation_info
else [
Document(page_content=doc["snippet"], metadata=doc) Document(page_content=doc["snippet"], metadata=doc)
for doc in response.generation_info["documents"] for doc in response.generation_info["documents"]
] ]
)
docs.append( docs.append(
Document( Document(
page_content=response.message.content, page_content=response.message.content,