File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed
tests/integration/test_knowledge_engines Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change 1
- """Core tests."""
1
+ """Core tests for reasoning with knowledge engines."""
2
+
2
3
import logging
3
4
import unittest
4
5
from typing import Iterator , List
@@ -145,7 +146,7 @@ def test_reason(self):
145
146
results = []
146
147
for task in self .tasks ():
147
148
result = reasoner .reason (task )
148
- print (yaml .dump (result .dict (), sort_keys = False ))
149
+ print (yaml .dump (result .model_dump (), sort_keys = False ))
149
150
print (result .prompt )
150
151
results .append (result )
151
152
ReasonerResultSet (results = [result ])
@@ -163,7 +164,7 @@ def test_reason_with_explanations(self):
163
164
task .include_explanations = True
164
165
result = reasoner .reason (task )
165
166
print (result .prompt )
166
- print (yaml .dump (result .dict (), sort_keys = False ))
167
+ print (yaml .dump (result .model_dump (), sort_keys = False ))
167
168
results .append (result )
168
169
for result in results :
169
170
print (
@@ -179,7 +180,7 @@ def test_reason_with_chain_of_thought(self):
179
180
task .chain_of_thought = True
180
181
result = reasoner .reason (task )
181
182
print (result .prompt )
182
- print (yaml .dump (result .dict (), sort_keys = False ))
183
+ print (yaml .dump (result .model_dump (), sort_keys = False ))
183
184
results .append (result )
184
185
for result in results :
185
186
print (
You can’t perform that action at this time.
0 commit comments