Spaces:
Sleeping
Sleeping
fix: align run_inference
Browse files
modules/info_extractor.py
CHANGED
|
@@ -21,10 +21,11 @@ class InfoExtractor:
|
|
| 21 |
# 确保调用AI模型时,使用类似 do_sample=False 或 temperature=0 的参数
|
| 22 |
# 这里我们模拟这个调用,并强调其重要性
|
| 23 |
log.info("🧠 使用LLM开始提取信息 (模式: 确定性)")
|
| 24 |
-
raw_response = self.ai_model.
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
|
|
|
| 28 |
)
|
| 29 |
|
| 30 |
try:
|
|
|
|
| 21 |
# 确保调用AI模型时,使用类似 do_sample=False 或 temperature=0 的参数
|
| 22 |
# 这里我们模拟这个调用,并强调其重要性
|
| 23 |
log.info("🧠 使用LLM开始提取信息 (模式: 确定性)")
|
| 24 |
+
raw_response = self.ai_model.run_inference(
|
| 25 |
+
input_type='text',
|
| 26 |
+
formatted_input=None,
|
| 27 |
+
prompt=prompt,
|
| 28 |
+
temperature=0.0
|
| 29 |
)
|
| 30 |
|
| 31 |
try:
|