Skip to content

Commit

Permalink
Merge pull request #303 from tvj15/fixing-comments-CohereCallResponse…
Browse files Browse the repository at this point in the history
…Chunk

fixed comments mentioned in issue #293
  • Loading branch information
brenkao committed Jun 6, 2024
2 parents bfc8c38 + 1ba10ef commit 3fc83e2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mirascope/cohere/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -286,7 +286,7 @@ def search_results(self) -> Optional[list[ChatSearchResult]]:

@property
def documents(self) -> Optional[list[ChatDocument]]:
"""Returns the documents for citation-generation event type else None."""
"""Returns the documents for search-results event type else None."""
if isinstance(self.chunk, StreamedChatResponse_SearchResults):
return self.chunk.documents
return None
Expand All @@ -300,7 +300,7 @@ def citations(self) -> Optional[list[ChatCitation]]:

@property
def response(self) -> Optional[NonStreamedChatResponse]:
"""Returns the response for text-generation event type else None."""
"""Returns the full response for the stream-end event type else None."""
if isinstance(self.chunk, StreamedChatResponse_StreamEnd):
return self.chunk.response
return None
Expand Down

0 comments on commit 3fc83e2

Please sign in to comment.