Skip to content

Commit

Permalink
feat: updated gdrive collection name to google!
Browse files Browse the repository at this point in the history
updated related parameters and variable names.
  • Loading branch information
amindadgar committed Jun 3, 2024
1 parent b327b70 commit 8bbf90c
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions subquery.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ def query_multiple_source(
community_id: str,
discord: bool = False,
discourse: bool = False,
gdrive: bool = False,
google: bool = False,
notion: bool = False,
telegram: bool = False,
github: bool = False,
Expand All @@ -44,7 +44,7 @@ def query_multiple_source(
discourse : bool
if `True` then add the engine to the subquery_generator
default is set to False
gdrive : bool
google : bool
if `True` then add the engine to the subquery_generator
default is set to False
notion : bool
Expand Down Expand Up @@ -73,7 +73,7 @@ def query_multiple_source(
discord_query_engine: BaseQueryEngine
github_query_engine: BaseQueryEngine
# discourse_query_engine: BaseQueryEngine
gdrive_query_engine: BaseQueryEngine
google_query_engine: BaseQueryEngine
notion_query_engine: BaseQueryEngine
mediawiki_query_engine: BaseQueryEngine
# telegram_query_engine: BaseQueryEngine
Expand Down Expand Up @@ -103,8 +103,8 @@ def query_multiple_source(

if discourse:
raise NotImplementedError
if gdrive and check_collection("gdrive"):
gdrive_query_engine = GDriveQueryEngine(community_id=community_id).prepare()
if google and check_collection("google"):
google_query_engine = GDriveQueryEngine(community_id=community_id).prepare()
tool_metadata = ToolMetadata(
name="Google-Drive",
description=(
Expand All @@ -114,7 +114,7 @@ def query_multiple_source(
)
query_engine_tools.append(
QueryEngineTool(
query_engine=gdrive_query_engine,
query_engine=google_query_engine,
metadata=tool_metadata,
)
)
Expand Down

0 comments on commit 8bbf90c

Please sign in to comment.