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

revert: Revert feat: Factor Implement Search Enhancement (#294) (#305)

Browse Source 
* Revert "feat: Factor Implement Search Enhancement (#294)"

This reverts commit 4ecf25f0acf2389a172b14d3dab20895daf2ab89.

* fix CI
This commit is contained in:
Xu Yang
2024-09-24 12:27:59 +08:00
committed by GitHub
parent e9e014f689
commit a0e9b10cc0
5 changed files with 2 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/factor_coder/CoSTEER/knowledge_management.py
+1 -43
View File
@@ -1,7 +1,6 @@
from __future__ import annotations
import copy
import heapq
import json
import random
import re
@@ -206,13 +205,11 @@ class FactorQueriedGraphKnowledge(FactorQueriedKnowledge):
former_traces: dict = {},
component_with_success_task: dict = {},
error_with_success_task: dict = {},
data_set_knowledge_dict: dict = {},
**kwargs,
) -> None:
self.former_traces = former_traces
self.component_with_success_task = component_with_success_task
self.error_with_success_task = error_with_success_task
self.data_set_knowledge_dict = data_set_knowledge_dict
super().__init__(**kwargs)
@@ -312,10 +309,6 @@ class FactorGraphRAGStrategy(RAGStrategy):
FACTOR_IMPLEMENT_SETTINGS.v2_query_error_limit,
knowledge_sampler=conf_knowledge_sampler,
)
factor_implementation_queried_graph_knowledge = self.dataset_query(
evo,
factor_implementation_queried_graph_knowledge,
)
return factor_implementation_queried_graph_knowledge
def analyze_component(
@@ -718,37 +711,9 @@ class FactorGraphRAGStrategy(RAGStrategy):
return factor_implementation_queried_graph_knowledge
def dataset_query(
self,
evo: EvolvableSubjects,
factor_implementation_queried_graph_knowledge: FactorQueriedGraphKnowledge,
) -> QueriedKnowledge | None:
for task_index, target_factor_task in enumerate(evo.sub_tasks):
target_factor_task_information = target_factor_task.get_task_information()
related_info = {}
knowledge_dict = self.knowledgebase.data_set_knowledge_dict
table_explanations = [f"{key}: {json.dumps(value)}" for key, value in knowledge_dict.items()]
similarity = calculate_embedding_distance_between_str_list(
[target_factor_task_information], table_explanations
)[0]
top_related_indexes = heapq.nlargest(10, range(len(similarity)), key=lambda i: similarity[i])
for index in top_related_indexes:
key = list(knowledge_dict.keys())[index]
related_info[key] = knowledge_dict[key]
factor_implementation_queried_graph_knowledge.data_set_knowledge_dict[
target_factor_task_information
] = related_info
return factor_implementation_queried_graph_knowledge
class FactorGraphKnowledgeBase(EvolvingKnowledgeBase):
def __init__(self, init_component_list=None, path: str | Path = None, data_set_knowledge_path=None) -> None:
def __init__(self, init_component_list=None, path: str | Path = None) -> None:
"""
Load knowledge, offer brief information of knowledge and common handle interfaces
"""
@@ -776,13 +741,6 @@ class FactorGraphKnowledgeBase(EvolvingKnowledgeBase):
# store the task description to component nodes
self.task_to_component_nodes = {}
# data set: data set information
self.data_set_knowledge_dict = {}
if data_set_knowledge_path:
with open(data_set_knowledge_path, "r") as f:
self.data_set_knowledge_dict = json.load(f)
super().__init__(path)
def get_all_nodes_by_label(self, label: str) -> list[UndirectedNode]:
return self.graph.get_all_nodes_by_label(label)