Skip to content

Commit 34207f9

Browse files
authored
Update COT datagen docs (#3610)
1 parent bb8cd65 commit 34207f9

File tree

2 files changed

+14
-14
lines changed

2 files changed

+14
-14
lines changed

docs/key_modules/datagen.md

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -47,25 +47,26 @@ The main class that implements the CoT generation system with the following capa
4747
from camel.datagen import CoTDataGenerator
4848

4949
# Initialize agents
50-
generator_agent = ChatAgent("System message for generator")
51-
verifier_agent = ChatAgent("System message for verifier")
50+
generator_agent = ChatAgent("Generator agent for simple math computation.")
51+
verifier_agent = ChatAgent("Verified agent for simple math computation.")
5252

5353
# Define golden answers
54+
question = "What's the answer of 1 + 2?"
55+
5456
golden_answers = {
55-
"question1": "answer1",
56-
"question2": "answer2"
57+
question: "3",
5758
}
5859

5960
# Create generator
6061
cot_generator = CoTDataGenerator(
6162
generator_agent=generator_agent,
6263
verifier_agent=verifier_agent,
6364
golden_answers=golden_answers,
64-
search_limit=100
65+
search_limit=3,
6566
)
6667

6768
# Generate solution
68-
solution = cot_generator.solve("question1")
69+
solution = cot_generator.solve(question)
6970
```
7071
</Card>
7172

@@ -115,7 +116,6 @@ The main class that implements the CoT generation system with the following capa
115116
<li>Solutions with intermediate steps</li>
116117
<li>Golden answers used for verification</li>
117118
<li>Export timestamp</li>
118-
<li>Solution scores and verification results</li>
119119
</ul>
120120
</Accordion>
121121
</AccordionGroup>

docs/mintlify/key_modules/datagen.mdx

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -47,25 +47,26 @@ The main class that implements the CoT generation system with the following capa
4747
from camel.datagen import CoTDataGenerator
4848

4949
# Initialize agents
50-
generator_agent = ChatAgent("System message for generator")
51-
verifier_agent = ChatAgent("System message for verifier")
50+
generator_agent = ChatAgent("Generator agent for simple math computation.")
51+
verifier_agent = ChatAgent("Verified agent for simple math computation.")
5252

5353
# Define golden answers
54+
question = "What's the answer of 1 + 2?"
55+
5456
golden_answers = {
55-
"question1": "answer1",
56-
"question2": "answer2"
57+
question: "3",
5758
}
5859

5960
# Create generator
6061
cot_generator = CoTDataGenerator(
6162
generator_agent=generator_agent,
6263
verifier_agent=verifier_agent,
6364
golden_answers=golden_answers,
64-
search_limit=100
65+
search_limit=3,
6566
)
6667

6768
# Generate solution
68-
solution = cot_generator.solve("question1")
69+
solution = cot_generator.solve(question)
6970
```
7071
</Card>
7172

@@ -115,7 +116,6 @@ The main class that implements the CoT generation system with the following capa
115116
<li>Solutions with intermediate steps</li>
116117
<li>Golden answers used for verification</li>
117118
<li>Export timestamp</li>
118-
<li>Solution scores and verification results</li>
119119
</ul>
120120
</Accordion>
121121
</AccordionGroup>

0 commit comments

Comments
 (0)