wehub-resource-sync
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
from rdagent.oai.llm_utils import APIBackend
|
|
from rdagent.utils.agent.tpl import T
|
|
|
|
|
|
def extract_knowledge_from_high_score_answers(content: str):
|
|
sys_prompt = T(".prompts:extract_kaggle_knowledge_prompts.system").r()
|
|
user_prompt = T(".prompts:extract_kaggle_knowledge_prompts.user").r(file_content=content)
|
|
|
|
response_analysis = APIBackend().build_messages_and_create_chat_completion(
|
|
user_prompt=user_prompt,
|
|
system_prompt=sys_prompt,
|
|
json_mode=True,
|
|
)
|
|
|
|
try:
|
|
response_json_analysis = json.loads(response_analysis)
|
|
except json.JSONDecodeError:
|
|
response_json_analysis = {"error": "Failed to parse LLM's response as JSON"}
|
|
|
|
return response_json_analysis
|
|
|
|
|
|
def extract_knowledge_from_feedback(feedback_response: dict) -> dict:
|
|
"""
|
|
Extracts knowledge from LLM-generated feedback and structures it.
|
|
"""
|
|
sys_prompt = T(".prompts:extract_kaggle_knowledge_from_feedback_prompts.system").r()
|
|
user_prompt = T(".prompts:extract_kaggle_knowledge_from_feedback_prompts.user").r(
|
|
experiment_strategy=feedback_response
|
|
)
|
|
|
|
response_analysis = APIBackend().build_messages_and_create_chat_completion(
|
|
user_prompt=user_prompt,
|
|
system_prompt=sys_prompt,
|
|
json_mode=True,
|
|
)
|
|
|
|
try:
|
|
response_json_analysis = json.loads(response_analysis)
|
|
except json.JSONDecodeError:
|
|
response_json_analysis = {"error": "Failed to parse LLM's response as JSON"}
|
|
|
|
return response_json_analysis
|
|
|
|
|
|
def process_all_case_files(directory_path: str):
|
|
output_file = Path(directory_path) / "kaggle_experience_results.json"
|
|
json_output = []
|
|
|
|
for file_path in Path(directory_path).rglob("*.case"):
|
|
with open(file_path, "r", encoding="utf-8") as file:
|
|
content = file.read()
|
|
knowledge = extract_knowledge_from_high_score_answers(content)
|
|
json_output.append(knowledge)
|
|
|
|
with open(output_file, "w", encoding="utf-8") as json_file:
|
|
json.dump(json_output, json_file, ensure_ascii=False)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
process_all_case_files(directory_path="git_ignore_folder/data/kaggle")
|