Update app.py
Browse files
app.py
CHANGED
|
@@ -1051,6 +1051,7 @@ if __name__ == '__main__':
|
|
| 1051 |
# 从环境变量加载词库文件路径,如未设置则使用默认路径
|
| 1052 |
wordpairs_file = os.getenv('WORDPAIRS_FILE', os.path.join(os.path.dirname(__file__), "all_word_pairs.json"))
|
| 1053 |
|
|
|
|
| 1054 |
agent = SpyAgent(name, model_name=os.getenv('MODEL_NAME'))
|
| 1055 |
|
| 1056 |
# 预先加载词库
|
|
@@ -1060,12 +1061,6 @@ if __name__ == '__main__':
|
|
| 1060 |
else:
|
| 1061 |
logger.warning(f"词库文件不存在: {wordpairs_file},将使用默认词库")
|
| 1062 |
|
| 1063 |
-
|
| 1064 |
-
|
| 1065 |
-
agent=SpyAgent(
|
| 1066 |
-
name,
|
| 1067 |
-
model_name=os.getenv('MODEL_NAME'),
|
| 1068 |
-
|
| 1069 |
-
)
|
| 1070 |
-
)
|
| 1071 |
agent_builder.start()
|
|
|
|
| 1051 |
# 从环境变量加载词库文件路径,如未设置则使用默认路径
|
| 1052 |
wordpairs_file = os.getenv('WORDPAIRS_FILE', os.path.join(os.path.dirname(__file__), "all_word_pairs.json"))
|
| 1053 |
|
| 1054 |
+
# 创建一个Agent对象
|
| 1055 |
agent = SpyAgent(name, model_name=os.getenv('MODEL_NAME'))
|
| 1056 |
|
| 1057 |
# 预先加载词库
|
|
|
|
| 1061 |
else:
|
| 1062 |
logger.warning(f"词库文件不存在: {wordpairs_file},将使用默认词库")
|
| 1063 |
|
| 1064 |
+
# 使用已加载词库的agent对象
|
| 1065 |
+
agent_builder = AgentBuilder(name, agent=agent)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1066 |
agent_builder.start()
|