mirror of
https://github.com/katanemo/plano.git
synced 2026-04-25 08:46:24 +02:00
ba651aaf71
* Rename all arch references to plano across the codebase
Complete rebrand from "Arch"/"archgw" to "Plano" including:
- Config files: arch_config_schema.yaml, workflow, demo configs
- Environment variables: ARCH_CONFIG_* → PLANO_CONFIG_*
- Python CLI: variables, functions, file paths, docker mounts
- Rust crates: config paths, log messages, metadata keys
- Docker/build: Dockerfile, supervisord, .dockerignore, .gitignore
- Docker Compose: volume mounts and env vars across all demos/tests
- GitHub workflows: job/step names
- Shell scripts: log messages
- Demos: Python code, READMEs, VS Code configs, Grafana dashboard
- Docs: RST includes, code comments, config references
- Package metadata: package.json, pyproject.toml, uv.lock
External URLs (docs.archgw.com, github.com/katanemo/archgw) left as-is.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* Update remaining arch references in docs
- Rename RST cross-reference labels: arch_access_logging, arch_overview_tracing, arch_overview_threading → plano_*
- Update label references in request_lifecycle.rst
- Rename arch_config_state_storage_example.yaml → plano_config_state_storage_example.yaml
- Update config YAML comments: "Arch creates/uses" → "Plano creates/uses"
- Update "the Arch gateway" → "the Plano gateway" in configuration_reference.rst
- Update arch_config_schema.yaml reference in provider_models.py
- Rename arch_agent_router → plano_agent_router in config example
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* Fix remaining arch references found in second pass
- config/docker-compose.dev.yaml: ARCH_CONFIG_FILE → PLANO_CONFIG_FILE,
arch_config.yaml → plano_config.yaml, archgw_logs → plano_logs
- config/test_passthrough.yaml: container mount path
- tests/e2e/docker-compose.yaml: source file path (was still arch_config.yaml)
- cli/planoai/core.py: comment and log message
- crates/brightstaff/src/tracing/constants.rs: doc comment
- tests/{e2e,archgw}/common.py: get_arch_messages → get_plano_messages,
arch_state/arch_messages variables renamed
- tests/{e2e,archgw}/test_prompt_gateway.py: updated imports and usages
- demos/shared/test_runner/{common,test_demos}.py: same renames
- tests/e2e/test_model_alias_routing.py: docstring
- .dockerignore: archgw_modelserver → plano_modelserver
- demos/use_cases/claude_code_router/pretty_model_resolution.sh: container name
Note: x-arch-* HTTP header values and Rust constant names intentionally
preserved for backwards compatibility with existing deployments.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---------
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
114 lines
3.5 KiB
Python
114 lines
3.5 KiB
Python
import json
|
|
import os
|
|
|
|
|
|
PROMPT_GATEWAY_ENDPOINT = os.getenv(
|
|
"PROMPT_GATEWAY_ENDPOINT", "http://localhost:10000/v1/chat/completions"
|
|
)
|
|
|
|
PROMPT_GATEWAY_PATH = os.getenv("PROMPT_GATEWAY_PATH", "/v1/chat/completions")
|
|
MODEL_SERVER_FUNC_PATH = os.getenv("MODEL_SERVER_FUNC_PATH", "/function_calling")
|
|
|
|
LLM_GATEWAY_ENDPOINT = os.getenv(
|
|
"LLM_GATEWAY_ENDPOINT", "http://localhost:12000/v1/chat/completions"
|
|
)
|
|
ARCH_STATE_HEADER = "x-arch-state"
|
|
|
|
PREFILL_LIST = [
|
|
"May",
|
|
"Could",
|
|
"Sure",
|
|
"Definitely",
|
|
"Certainly",
|
|
"Of course",
|
|
"Can",
|
|
]
|
|
|
|
TEST_CASE_FIXTURES = {
|
|
"SIMPLE": {
|
|
"input": {
|
|
"messages": [
|
|
{
|
|
"role": "user",
|
|
"content": "how is the weather in seattle for next 2 days",
|
|
}
|
|
]
|
|
},
|
|
"model_server_response": {
|
|
"id": 0,
|
|
"object": "chat_completion",
|
|
"created": "",
|
|
"choices": [
|
|
{
|
|
"id": 0,
|
|
"message": {
|
|
"role": "",
|
|
"content": "",
|
|
"tool_call_id": "",
|
|
"tool_calls": [
|
|
{
|
|
"id": "call_6009",
|
|
"type": "function",
|
|
"function": {
|
|
"name": "get_current_weather",
|
|
"arguments": {
|
|
"location": "Seattle, WA",
|
|
"days": "2",
|
|
},
|
|
},
|
|
}
|
|
],
|
|
},
|
|
"finish_reason": "stop",
|
|
}
|
|
],
|
|
"model": "Arch-Function",
|
|
"metadata": {"intent_latency": "455.092", "function_latency": "312.744"},
|
|
},
|
|
"api_server_response": [
|
|
{
|
|
"date": "2024-12-12",
|
|
"temperature": {"min": 72, "max": 90},
|
|
"units": "Farenheit",
|
|
"query_time": "2024-12-12 22:06:30.420319+00:00",
|
|
},
|
|
{
|
|
"date": "2024-12-13",
|
|
"temperature": {"min": 52, "max": 70},
|
|
"units": "Farenheit",
|
|
"query_time": "2024-12-12 22:06:30.420349+00:00",
|
|
},
|
|
],
|
|
}
|
|
}
|
|
|
|
|
|
def get_data_chunks(stream, n=1):
|
|
chunks = []
|
|
for chunk in stream.iter_lines():
|
|
if chunk:
|
|
chunk = chunk.decode("utf-8")
|
|
chunk_data_id = chunk[0:6]
|
|
assert chunk_data_id == "data: "
|
|
chunk_data = chunk[6:]
|
|
chunk_data = chunk_data.strip()
|
|
chunks.append(chunk_data)
|
|
if len(chunks) >= n:
|
|
break
|
|
return chunks
|
|
|
|
|
|
def get_plano_messages(response_json):
|
|
plano_messages = []
|
|
if response_json and "metadata" in response_json:
|
|
# load plano_state from metadata
|
|
plano_state_str = response_json.get("metadata", {}).get(ARCH_STATE_HEADER, "{}")
|
|
# parse plano_state into json object
|
|
plano_state = json.loads(plano_state_str)
|
|
# load messages from plano_state
|
|
plano_messages_str = plano_state.get("messages", "[]")
|
|
# parse messages into json object
|
|
plano_messages = json.loads(plano_messages_str)
|
|
# append messages from plano gateway to history
|
|
return plano_messages
|
|
return []
|