Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feat/327 use hivemind backend lib #328

Merged
merged 7 commits into from
Nov 21, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion dags/analyzer_helper/common/fetch_platforms.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from bson import ObjectId
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class FetchPlatforms:
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/common/load_transformed_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from analyzer_helper.common.base.load_transformed_data_base import (
LoadTransformedDataBase,
)
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class LoadTransformedData(LoadTransformedDataBase):
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/common/load_transformed_members.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from analyzer_helper.common.base.load_transformed_members_base import (
LoadTransformedMembersBase,
)
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class LoadTransformedMembers(LoadTransformedMembersBase):
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/discord/discord_extract_raw_infos.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from datetime import datetime, timezone

from analyzer_helper.discord.extract_raw_info_base import ExtractRawInfosBase
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class DiscordExtractRawInfos(ExtractRawInfosBase):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import logging

from analyzer_helper.discord.extract_raw_member_base import ExtractRawMembersBase
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class DiscordExtractRawMembers(ExtractRawMembersBase):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import logging

from analyzer_helper.discord.load_transformed_data_base import LoadTransformedDataBase
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class DiscordLoadTransformedData(LoadTransformedDataBase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from analyzer_helper.discord.load_transformed_members_base import (
LoadTransformedMembersBase,
)
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class DiscordLoadTransformedMembers(LoadTransformedMembersBase):
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/discord/fetch_discord_platforms.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from bson import ObjectId
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton
amindadgar marked this conversation as resolved.
Show resolved Hide resolved


class FetchDiscordPlatforms:
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/discord/utils/is_user_bot.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class UserBotChecker:
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/discourse/extract_raw_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
DateTimeFormatConverter,
)
from github.neo4j_storage.neo4j_connection import Neo4jConnection
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class ExtractRawInfo:
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/discourse/extract_raw_members.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
DateTimeFormatConverter,
)
from github.neo4j_storage.neo4j_connection import Neo4jConnection
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class ExtractRawMembers:
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/telegram/extract_raw_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
DateTimeFormatConverter,
)
from github.neo4j_storage.neo4j_connection import Neo4jConnection
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class ExtractRawInfo:
Expand Down
2 changes: 1 addition & 1 deletion dags/analyzer_helper/telegram/extract_raw_members.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
DateTimeFormatConverter,
)
from github.neo4j_storage.neo4j_connection import Neo4jConnection
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from pymongo import DESCENDING
from tc_hivemind_backend.db.mongo import MongoSingleton


class ExtractRawMembers:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from analyzer_helper.telegram.extract_raw_data import ExtractRawInfo
from github.neo4j_storage.neo4j_connection import Neo4jConnection
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestExtractRawInfo(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from analyzer_helper.discord.discord_extract_raw_infos import DiscordExtractRawInfos
from bson import ObjectId
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestDiscordExtractRawInfos(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from datetime import datetime

from analyzer_helper.discord.discord_extract_raw_members import DiscordExtractRawMembers
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestDiscordExtractRawMembers(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from analyzer_helper.discord.utils.is_user_bot import UserBotChecker
from bson import ObjectId
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestUserBotChecker(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from analyzer_helper.discord.discord_load_transformed_data import (
DiscordLoadTransformedData,
)
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestDiscordLoadTransformedData(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from analyzer_helper.discord.discord_load_transformed_members import (
DiscordLoadTransformedMembers,
)
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestDiscordLoadTransformedMembers(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from analyzer_helper.discord.discord_transform_raw_data import DiscordTransformRawData
from analyzer_helper.discord.utils.is_user_bot import UserBotChecker
from bson import ObjectId
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestDiscordTransformRawData(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from analyzer_helper.discourse.extract_raw_data import ExtractRawInfo
from github.neo4j_storage.neo4j_connection import Neo4jConnection
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestExtractRawInfo(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from analyzer_helper.discord.fetch_discord_platforms import FetchDiscordPlatforms
from bson import ObjectId
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestFetchDiscordPlatforms(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

from analyzer_helper.discord.fetch_discord_platforms import FetchDiscordPlatforms
from bson import ObjectId
from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestFetchDiscordPlatformsUnit(unittest.TestCase):
Expand Down
2 changes: 1 addition & 1 deletion dags/hivemind_etl_helpers/github_etl.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
from datetime import datetime

from dotenv import load_dotenv
from hivemind_etl_helpers.ingestion_pipeline import CustomIngestionPipeline
from hivemind_etl_helpers.src.db.github.extract import (
GithubExtraction,
fetch_issues,
Expand All @@ -13,6 +12,7 @@
)
from hivemind_etl_helpers.src.db.github.transform import GitHubTransformation
from llama_index.core import Document
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline


def process_github_vectorstore(
Expand Down
Loading