From 1dd69afa0c94213d9853287285179defec3e5665 Mon Sep 17 00:00:00 2001 From: ucloudnb666 Date: Sat, 9 May 2026 17:20:03 +0800 Subject: [PATCH 1/3] feat: add Astraflow provider to griptape/drivers/prompt/astraflow_prompt_driver.py --- .../drivers/prompt/astraflow_prompt_driver.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 griptape/drivers/prompt/astraflow_prompt_driver.py diff --git a/griptape/drivers/prompt/astraflow_prompt_driver.py b/griptape/drivers/prompt/astraflow_prompt_driver.py new file mode 100644 index 000000000..c3663b186 --- /dev/null +++ b/griptape/drivers/prompt/astraflow_prompt_driver.py @@ -0,0 +1,17 @@ +from __future__ import annotations + +from attrs import define, field + +from griptape.drivers.prompt.openai import OpenAiChatPromptDriver + + +@define +class AstraflowPromptDriver(OpenAiChatPromptDriver): + """Prompt Driver for Astraflow (UCloud / 优刻得). + + Astraflow is an OpenAI-compatible AI model aggregation platform supporting 200+ models. + Set ``base_url`` to ``https://api.modelverse.cn/v1`` and use ``ASTRAFLOW_CN_API_KEY`` + to use the China endpoint instead of the global endpoint. + """ + + base_url: str = field(default="https://api-us-ca.umodelverse.ai/v1", kw_only=True, metadata={"serializable": True}) From daffe15d6bc4bb5f9499fc450a79307fa1b7fd02 Mon Sep 17 00:00:00 2001 From: ucloudnb666 Date: Sat, 9 May 2026 17:20:05 +0800 Subject: [PATCH 2/3] feat: add Astraflow provider to griptape/drivers/prompt/astraflow/__init__.py --- griptape/drivers/prompt/astraflow/__init__.py | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 griptape/drivers/prompt/astraflow/__init__.py diff --git a/griptape/drivers/prompt/astraflow/__init__.py b/griptape/drivers/prompt/astraflow/__init__.py new file mode 100644 index 000000000..830ad8110 --- /dev/null +++ b/griptape/drivers/prompt/astraflow/__init__.py @@ -0,0 +1,5 @@ +from griptape.drivers.prompt.astraflow_prompt_driver import AstraflowPromptDriver + +__all__ = [ + "AstraflowPromptDriver", +] From 0ee4244e1ca0c786c731a6e37e780dc58a891f30 Mon Sep 17 00:00:00 2001 From: ucloudnb666 Date: Sat, 9 May 2026 17:20:07 +0800 Subject: [PATCH 3/3] feat: add Astraflow provider to griptape/drivers/__init__.py --- griptape/drivers/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/griptape/drivers/__init__.py b/griptape/drivers/__init__.py index f95aba1cc..64cd16c80 100644 --- a/griptape/drivers/__init__.py +++ b/griptape/drivers/__init__.py @@ -14,6 +14,7 @@ from .prompt.dummy import DummyPromptDriver from .prompt.ollama import OllamaPromptDriver from .prompt.grok import GrokPromptDriver +from .prompt.astraflow import AstraflowPromptDriver from .prompt.griptape_cloud import GriptapeCloudPromptDriver from .prompt.perplexity import PerplexityPromptDriver @@ -144,6 +145,7 @@ "AmazonSageMakerJumpstartEmbeddingDriver", "AmazonSageMakerJumpstartPromptDriver", "AmazonSqsEventListenerDriver", + "AstraflowPromptDriver", "AnthropicPromptDriver", "AstraDbVectorStoreDriver", "AwsIotCoreEventListenerDriver",