From 53e7707bc7810486d2457821ac7c43cf3b76c05b Mon Sep 17 00:00:00 2001
From: steveh8758_lab <steveh8758@gmail.com>
Date: Sun, 23 Feb 2025 20:11:32 +0800
Subject: [PATCH] SEE desc

MOD:
    to let user setting config.ini
---
 config.ini       |  6 ++++++
 main.py          | 20 +++++++++++++-------
 requirements.txt |  8 +++++++-
 3 files changed, 26 insertions(+), 8 deletions(-)
 create mode 100644 config.ini

diff --git a/config.ini b/config.ini
new file mode 100644
index 0000000..c0fe018
--- /dev/null
+++ b/config.ini
@@ -0,0 +1,6 @@
+[MAIN]
+is_local = False
+provider_name = ollama
+provider_model = deepseek-r1:32b
+provider_server_address = 127.0.0.1:6666
+agent_name = jarvis
diff --git a/main.py b/main.py
index ec2561b..8d09851 100755
--- a/main.py
+++ b/main.py
@@ -3,6 +3,7 @@
 import sys
 import signal
 import argparse
+import configparser
 
 from sources.llm_provider import Provider
 from sources.interaction import Interaction
@@ -14,21 +15,26 @@ parser.add_argument('--speak', action='store_true',
                 help='Make AI use text-to-speech')
 args = parser.parse_args()
 
+config = configparser.ConfigParser()
+config.read('config.ini')
+
 def handleInterrupt(signum, frame):
     sys.exit(0)
 
 def main():
     signal.signal(signal.SIGINT, handler=handleInterrupt)
 
-    #local_provider = Provider("ollama", "deepseek-r1:14b", "127.0.0.1:5000")
-    server_provider = Provider(provider_name="server",
-                               model="deepseek-r1:14b",
-                               server_address="192.168.1.100:5000")
+    if config.getboolean('MAIN', 'is_local'):
+        provider = Provider(config["MAIN"]["provider_name"], config["MAIN"]["provider_model"], config["MAIN"]["provider_server_address"])
+    else:
+        provider = Provider(provider_name=config["MAIN"]["provider_name"],
+                                   model=config["MAIN"]["provider_model"],
+                                   server_address=config["MAIN"]["provider_server_address"])
 
-    agent = CoderAgent(model="deepseek-r1:14b",
-                       name="jarvis",
+    agent = CoderAgent(model=config["MAIN"]["provider_model"],
+                       name=config["MAIN"]["agent_name"],
                        prompt_path="prompts/coder_agent.txt",
-                       provider=server_provider)
+                       provider=provider)
 
     interaction = Interaction([agent], tts_enabled=args.speak)
     while interaction.is_active:
diff --git a/requirements.txt b/requirements.txt
index a0f0eae..e79edcc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,4 +11,10 @@ scipy==1.15.1
 kokoro==0.7.12
 flask==3.1.0
 soundfile==0.13.1
-termcolor
\ No newline at end of file
+termcolor
+
+# if use chinese
+ordered_set
+pypinyin
+cn2an
+jieba
\ No newline at end of file