chewt/ai-assistant
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Fix --host argument

Browse source
This commit is contained in:
Hayden Johnson 2024-09-25 13:02:02 -07:00
parent 8deaa3b6e1
commit a69b63e4fe
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
assistant.py
View file

@ -88,8 +88,6 @@ def copy_string_to_clipboard(string):
except: except:
return return
# Point to the local server
client = Client(host=server)
code_history = [] code_history = []
@ -151,8 +149,8 @@ def parse_args():
parser.add_argument('--model', '-m', nargs='?', const=True, default=False, help='Specify model') parser.add_argument('--model', '-m', nargs='?', const=True, default=False, help='Specify model')
# Add the --temp (-t) argument # Add the --temp (-t) argument
parser.add_argument('--temp', '-t', nargs='?', const=True, default=False, help='Specify temperature') parser.add_argument('--temp', '-t', nargs='?', const=True, default=False, help='Specify temperature')
# Add the --host (-s) argument # Add the --host (-h) argument
parser.add_argument('--h', '-h', nargs='1', const=True, default=False, help='Specify host of ollama server') parser.add_argument('--host', nargs='?', const=True, default=False, help='Specify host of ollama server')
# Parse the arguments # Parse the arguments
return parser.parse_args() return parser.parse_args()
@ -210,10 +208,15 @@ def handle_non_piped_input(args):
else: else:
chat(user_input) chat(user_input)
client = None
def main(): def main():
args = parse_args() args = parse_args()
if args.host: if args.host:
set_host(args.host) set_host(args.host)
# Point to the local server
global client
client = Client(host=server)
if args.model: if args.model:
global model global model
model = args.model model = args.model