Skip to content

Commit

Permalink
fix type error
Browse files Browse the repository at this point in the history
  • Loading branch information
ricopinazo committed May 20, 2024
1 parent 99ad149 commit 7b4b183
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions src/comai/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import typer
import click
import itertools
from typing import List, Optional, Iterator
from typing import List, Optional, Iterator, Literal
from typing_extensions import Annotated
from langchain_community.llms.ollama import OllamaEndpointNotFoundError
from urllib3.exceptions import NewConnectionError
Expand Down Expand Up @@ -76,10 +76,8 @@ def settings_callback(value: bool):
else:
raise Exception("No models available for the selected provider")
model = prompt_options("Which model do you want to use?", models, default_model)

settings.provider = provider
settings.model = model
write_settings(settings)
updated_settings = Settings.parse_obj({"provider": provider, "model": model})
write_settings(updated_settings)
raise typer.Exit()


Expand Down

0 comments on commit 7b4b183

Please sign in to comment.