Skip to content

feat: add Astraflow prompt driver (OpenAI-compatible)#2169

Open
ucloudnb666 wants to merge 3 commits into
griptape-ai:mainfrom
ucloudnb666:feat/astraflow-1778318397
Open

feat: add Astraflow prompt driver (OpenAI-compatible)#2169
ucloudnb666 wants to merge 3 commits into
griptape-ai:mainfrom
ucloudnb666:feat/astraflow-1778318397

Conversation

@ucloudnb666

Copy link
Copy Markdown

Summary

Adds support for Astraflow (by UCloud / 优刻得), an OpenAI-compatible AI model aggregation platform supporting 200+ models.

What's added

  • griptape/drivers/prompt/astraflow_prompt_driver.pyAstraflowPromptDriver, a thin subclass of OpenAiChatPromptDriver that defaults base_url to Astraflow's global endpoint (https://api-us-ca.umodelverse.ai/v1).
  • griptape/drivers/prompt/astraflow/__init__.py — provider subpackage following the existing convention (mirrors the layout used by grok, perplexity, etc.).
  • Registers AstraflowPromptDriver in the legacy griptape.drivers re-export module for parity with other providers.

Why a thin subclass?

Astraflow exposes an OpenAI-compatible REST API, so — like the existing GrokPromptDriver and PerplexityPromptDriver — the integration is just a different base_url. No new SDK or transport code is needed.

Usage

import os
from griptape.drivers.prompt.astraflow import AstraflowPromptDriver

# Global endpoint (default)
driver = AstraflowPromptDriver(
    api_key=os.environ["ASTRAFLOW_API_KEY"],
    model="gpt-4o-mini",
)

# China endpoint
driver = AstraflowPromptDriver(
    api_key=os.environ["ASTRAFLOW_CN_API_KEY"],
    base_url="https://api.modelverse.cn/v1",
    model="gpt-4o-mini",
)

Endpoints

Region Base URL API key env var
Global https://api-us-ca.umodelverse.ai/v1 ASTRAFLOW_API_KEY
China https://api.modelverse.cn/v1 ASTRAFLOW_CN_API_KEY

@ucloudnb666 ucloudnb666 changed the title Add Astraflow prompt driver (OpenAI-compatible) feat: add Astraflow prompt driver (OpenAI-compatible) May 13, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant