Skip to content

Commit 4966bdd

Browse files
committed
support both ollama and openai LLMs
1 parent 0f7c62e commit 4966bdd

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/command_tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
def main():
1212
config = Config() # 创建配置实例
1313
github_client = GitHubClient(config.github_token) # 创建GitHub客户端实例
14-
llm = LLM() # 创建语言模型实例
14+
llm = LLM(config) # 创建语言模型实例
1515
report_generator = ReportGenerator(llm) # 创建报告生成器实例
1616
subscription_manager = SubscriptionManager(config.subscriptions_file) # 创建订阅管理器实例
1717
command_handler = CommandHandler(github_client, subscription_manager, report_generator) # 创建命令处理器实例

src/gradio_server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
# 创建各个组件的实例
1111
config = Config()
1212
github_client = GitHubClient(config.github_token)
13-
llm = LLM()
13+
llm = LLM(config) # 创建语言模型实例
1414
report_generator = ReportGenerator(llm)
1515
subscription_manager = SubscriptionManager(config.subscriptions_file)
1616

0 commit comments

Comments
 (0)