DiffusionCurriculum

Install

按照uv安装

git clone https://github.com/shijian2001/DiffusionCurriculum
cd DiffusionCurriculum
uv sync
source .venv/bin/activate

Note：必须按照上述方法安装，不然无法正常使用我们自己开发的package

QA API 调用

参见scripts/test_api.py

Note: 如果需要解析llm输出的json格式结果，可以使用：

from src.utils import JSONParser

results = JSONParser.parse("your response")

如果想检验自己输出的格式或者有更复杂的验证逻辑，可以定义validate_func(answer:str)->Any,若没有则传入None。其中validate函数的返回应该是通过验证的结果或False。切记不能返回True:

async for result in generator.generate_stream(prompts, system_prompt, validate_func=validate_func):
    pass

Dev Process

参照relation generation的开发流程：

1. 在src/scene_graph_builder/relation_generator.py中开发核心功能
2. 在scripts/generate_relations开发脚本
3. 脚本传参完全通过configs/relation_gen.yaml控制，执行:

python scripts/generate_relations.py configs/relation_gen.yaml

如果你有新添加的package，请在pyproject.toml中的dependence中添加：

dependencies = [
    "openai>=0.28.0",
    "PyYAML>=6.0",
    ... # Add
]