highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-05 17:53:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

mypy fixes

Browse Source
This commit is contained in:
Weves
2024-02-27 15:54:19 -08:00
committed by Chris Weaver
parent 3941111685
commit cd198ba368
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/danswer/connectors/document360/connector.py
View File

@@ -17,6 +17,7 @@ from danswer.connectors.interfaces import GenerateDocumentsOutput
from danswer.connectors.interfaces import LoadConnector from danswer.connectors.interfaces import LoadConnector
from danswer.connectors.interfaces import PollConnector from danswer.connectors.interfaces import PollConnector
from danswer.connectors.interfaces import SecondsSinceUnixEpoch from danswer.connectors.interfaces import SecondsSinceUnixEpoch
from danswer.connectors.models import BasicExpertInfo
from danswer.connectors.models import ConnectorMissingCredentialError from danswer.connectors.models import ConnectorMissingCredentialError
from danswer.connectors.models import Document from danswer.connectors.models import Document
from danswer.connectors.models import Section from danswer.connectors.models import Section
@@ -130,7 +131,9 @@ class Document360Connector(LoadConnector, PollConnector):
continue continue
authors = [ authors = [
{"name": author.get("name", "Unknown"), "email": author["email_id"]} BasicExpertInfo(
display_name=author.get("name"), email=author["email_id"]
)
for author in article_details.get("authors", []) for author in article_details.get("authors", [])
if author["email_id"] if author["email_id"]
] ]

2
backend/danswer/search/search_runner.py
View File

@@ -249,7 +249,7 @@ def semantic_reranking(
rerank_metrics_callback( rerank_metrics_callback(
RerankMetricsContainer( RerankMetricsContainer(
metrics=chunk_metrics, raw_similarity_scores=ranked_raw_scores metrics=chunk_metrics, raw_similarity_scores=ranked_raw_scores # type: ignore
) )
) )