gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-03 02:17:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Initial version if Graph RAG in KAGGLE scenario (#301)

Browse Source 
* Initial version if Graph RAG in KAGGLE scenario

* fix CI

* fix a small bug

* fix CI

* fix CI

* fix CI
This commit is contained in:
Xu Yang
2024-09-23 19:32:01 +08:00
committed by GitHub
parent 972bb2d99f
commit 003b067499
22 changed files with 382 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/core/evolving_framework.py
+4 -2
View File
@@ -5,6 +5,8 @@ from abc import ABC, abstractmethod
from dataclasses import dataclass
from typing import TYPE_CHECKING, Any
from rdagent.core.knowledge_base import KnowledgeBase
if TYPE_CHECKING:
from rdagent.core.evaluation import Feedback
from rdagent.core.scenario import Scenario
@@ -18,7 +20,7 @@ class QueriedKnowledge:
pass
class KnowledgeBase(ABC):
class EvolvingKnowledgeBase(KnowledgeBase):
@abstractmethod
def query(
self,
@@ -78,7 +80,7 @@ class EvolvingStrategy(ABC):
class RAGStrategy(ABC):
"""Retrieval Augmentation Generation Strategy"""
def __init__(self, knowledgebase: KnowledgeBase) -> None:
def __init__(self, knowledgebase: EvolvingKnowledgeBase) -> None:
self.knowledgebase = knowledgebase
@abstractmethod