rune/oai
1
0
Fork 0
Code Issues Pull Requests Releases 2 Activity

Small changes

Browse Source
This commit is contained in:
Rune Olsen
2025-12-04 15:02:37 +01:00
parent 8aede55362
commit 028d2911c3
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
oai.py
View File

@@ -70,7 +70,7 @@ except Exception as e:
models_data = []
text_models = []
# **NEW: Function to clear the screen using ANSI escape sequences (cross-platform)**
# Function to clear the screen using ANSI escape sequences (cross-platform)
def clear_screen():
"""Clear the terminal screen using ANSI escape codes, with fallback to newlines."""
try:
@@ -112,12 +112,11 @@ def chat():
if not selected_model:
console.print("[bold yellow]No model selected. Use '/model' to choose one.[/]")
# NEW: Initialize PromptSession for input history and arrow keys
# Initialize PromptSession for input history and arrow keys
session = PromptSession(history=None) # 'None' uses InMemoryHistory by default; can be customized later for disk persistence
while True:
try:
# REPLACED: Use session.prompt() instead of typer.prompt() for history/arrow key support
user_input = session.prompt("You> ").strip()
if user_input.lower() in ["exit", "quit", "bye"]:
console.print("[bold yellow]Goodbye![/]")
@@ -195,7 +194,7 @@ def chat():
table.add_row("Base URL", OPENROUTER_BASE_URL or "[Not set]")
table.add_row("Streaming", "Enabled" if STREAM_ENABLED == "on" else "Disabled")
table.add_row("Database", str(database) or "[Not set]")
table.add_row("Current Model", str(selected_model["name"]) or "[Not set]")
table.add_row("Current Model", "[Not set]" if selected_model is None else str(selected_model["name"]))
console.print(Panel(table, title="[bold green]Current Configurations[/]", title_align="left"))
continue
@@ -233,7 +232,7 @@ def chat():
"View all current configurations.",
"/config\n(Displays table of API Key, Base URL, etc.)"
)
help_table.add_row( # **NEW: Added /clear to help**
help_table.add_row(
"/clear",
"Clear the terminal screen for a clean interface.",
"/clear\n[bold cyan]Screen cleared. Ready for your next input![/bold cyan]"
@@ -367,11 +366,11 @@ def chat():
console.print("[bold red]No response received.[/]")
except KeyboardInterrupt:
# NEW: Handle Ctrl+C during prompt input (continue loop instead of crashing)
# Handle Ctrl+C during prompt input (continue loop instead of crashing)
console.print("\n[bold yellow]Input interrupted. Continuing...[/]")
continue
except EOFError:
# NEW: Handle Ctrl+D (exit loop gracefully)
# Handle Ctrl+D (exit loop gracefully)
console.print("\n[bold yellow]Goodbye![/]")
return
except Exception as e: