Skip to content

Set token before connection probing #196

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
simo-prior merged 4 commits into from
Dec 12, 2025
Merged

Set token before connection probing #196

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
e2a0665
Set token before connection probing
simo-prior Dec 11, 2025
3ae835a
raise connection error on try_reuse_existing_token()
simo-prior Dec 12, 2025
cb1199f
try tests
simo-prior Dec 12, 2025
d6b67f3
fix tests
simo-prior Dec 12, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 16 additions & 8 deletions src/tabpfn_client/config.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,20 @@

import shutil

from httpx import ConnectError

from tabpfn_client.client import ServiceClient
from tabpfn_client.service_wrapper import UserAuthenticationClient
from tabpfn_client.constants import CACHE_DIR
from tabpfn_client.prompt_agent import PromptAgent
from tabpfn_client.ui import console, warn


CONNECTION_ERROR = RuntimeError(
"TabPFN is inaccessible at the moment, please try again later."
)


class Config:
def __new__(cls, *args, **kwargs):
"""
Expand All @@ -31,16 +38,17 @@ def init(use_server=True):
return

if use_server:
# check connection to server
if not UserAuthenticationClient.is_accessible_connection():
raise RuntimeError(
"TabPFN is inaccessible at the moment, please try again later."
try:
is_valid_token, access_token = (
UserAuthenticationClient.try_reuse_existing_token()
)
except ConnectError:
raise CONNECTION_ERROR

(
is_valid_token,
access_token,
) = UserAuthenticationClient.try_reuse_existing_token()
# TODO: no need to check connection again if token is valid, need to
# adjust tests accordingly.
if not UserAuthenticationClient.is_accessible_connection():
raise CONNECTION_ERROR

if is_valid_token:
PromptAgent.prompt_reusing_existing_token()
Expand Down