Merge pull request #7 from cubenlp/rex/option

fix type of options

askchat/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Rex Wang"""
 __email__ = '[email protected]'
-__version__ = '1.1.3'
+__version__ = '1.1.4'
 
 import asyncio
 from pathlib import Path

askchat/cli.py

@@ -203,7 +203,19 @@ def main( message, model, base_url, api_base, api_key, use_env
             click.echo("Please specify message!")
             return
         chat.user(message_text)
-    # Add chat response
+    # additinal options
+    if option:
+        option = dict(option)
+        ints = ['max_tokens', 'n', 'top_logprobs', 'seed']
+        floats = ['temperature', 'presence_penalty', 'frequency_penalty', 'top_p']
+        for key, value in option.items():
+            if key in ints:
+                option[key] = int(value)
+            elif key in floats:
+                option[key] = float(value)
+    else:
+        option = {}
+    # Add chat responses
     chat.assistant(asyncio.run(show_resp(chat, **dict(option))))
     chat.save(LAST_CHAT_FILE, mode='w')
 

setup.py

@@ -4,7 +4,7 @@
 
 from setuptools import setup, find_packages
 
-VERSION = '1.1.3'
+VERSION = '1.1.4'
 
 with open('README.md') as readme_file:
     readme = readme_file.read()