Skip to content
Open
Changes from 1 commit
Commits
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
4 changes: 3 additions & 1 deletion src/seclab_taskflow_agent/capi.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,8 @@ def check_tool_calls(self, _model: str, model_info: dict) -> bool:
),
}

_DEFAULT_PROVIDER = "api.githubcopilot.com"

def get_provider(endpoint: str | None = None) -> APIProvider:
"""Return the ``APIProvider`` for the given (or configured) endpoint URL.
Expand Down Expand Up @@ -180,7 +182,7 @@ def get_provider(endpoint: str | None = None) -> APIProvider:

def get_AI_endpoint() -> str:
"""Return the configured AI API endpoint URL."""
return os.getenv("AI_API_ENDPOINT", default="https://models.github.ai/inference")
return os.getenv("AI_API_ENDPOINT", default=_PROVIDERS[_DEFAULT_PROVIDER].base_url)
Comment on lines 183 to +185


def get_AI_token() -> str:
Expand Down
Loading