mirror of
https://github.com/catlog22/Claude-Code-Workflow.git
synced 2026-02-09 02:24:11 +08:00
bf66b095c7
- Create ccw-litellm Python package with AbstractEmbedder and AbstractLLMClient interfaces - Add BaseEmbedder abstraction and factory pattern to codex-lens for pluggable backends - Implement API Settings dashboard page for provider credentials and custom endpoints - Add REST API routes for CRUD operations on providers and endpoints - Extend CLI with --model parameter for custom endpoint routing - Integrate existing context-cache for @pattern file resolution - Add provider model registry with predefined models per provider type - Include i18n translations (en/zh) for all new UI elements 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
109 lines
3.0 KiB
Python
109 lines
3.0 KiB
Python
"""CLI entry point for ccw-litellm."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def main() -> int:
|
|
"""Main CLI entry point."""
|
|
parser = argparse.ArgumentParser(
|
|
prog="ccw-litellm",
|
|
description="Unified LiteLLM interface for ccw and codex-lens",
|
|
)
|
|
subparsers = parser.add_subparsers(dest="command", help="Available commands")
|
|
|
|
# config command
|
|
config_parser = subparsers.add_parser("config", help="Show configuration")
|
|
config_parser.add_argument(
|
|
"--path",
|
|
type=Path,
|
|
help="Configuration file path",
|
|
)
|
|
|
|
# embed command
|
|
embed_parser = subparsers.add_parser("embed", help="Generate embeddings")
|
|
embed_parser.add_argument("texts", nargs="+", help="Texts to embed")
|
|
embed_parser.add_argument(
|
|
"--model",
|
|
default="default",
|
|
help="Embedding model name (default: default)",
|
|
)
|
|
embed_parser.add_argument(
|
|
"--output",
|
|
choices=["json", "shape"],
|
|
default="shape",
|
|
help="Output format (default: shape)",
|
|
)
|
|
|
|
# chat command
|
|
chat_parser = subparsers.add_parser("chat", help="Chat with LLM")
|
|
chat_parser.add_argument("message", help="Message to send")
|
|
chat_parser.add_argument(
|
|
"--model",
|
|
default="default",
|
|
help="LLM model name (default: default)",
|
|
)
|
|
|
|
# version command
|
|
subparsers.add_parser("version", help="Show version")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.command == "version":
|
|
from . import __version__
|
|
|
|
print(f"ccw-litellm {__version__}")
|
|
return 0
|
|
|
|
if args.command == "config":
|
|
from .config import get_config
|
|
|
|
try:
|
|
config = get_config(config_path=args.path if hasattr(args, "path") else None)
|
|
print(config.model_dump_json(indent=2))
|
|
except Exception as e:
|
|
print(f"Error loading config: {e}", file=sys.stderr)
|
|
return 1
|
|
return 0
|
|
|
|
if args.command == "embed":
|
|
from .clients import LiteLLMEmbedder
|
|
|
|
try:
|
|
embedder = LiteLLMEmbedder(model=args.model)
|
|
vectors = embedder.embed(args.texts)
|
|
|
|
if args.output == "json":
|
|
print(json.dumps(vectors.tolist()))
|
|
else:
|
|
print(f"Shape: {vectors.shape}")
|
|
print(f"Dimensions: {embedder.dimensions}")
|
|
except Exception as e:
|
|
print(f"Error: {e}", file=sys.stderr)
|
|
return 1
|
|
return 0
|
|
|
|
if args.command == "chat":
|
|
from .clients import LiteLLMClient
|
|
from .interfaces import ChatMessage
|
|
|
|
try:
|
|
client = LiteLLMClient(model=args.model)
|
|
response = client.chat([ChatMessage(role="user", content=args.message)])
|
|
print(response.content)
|
|
except Exception as e:
|
|
print(f"Error: {e}", file=sys.stderr)
|
|
return 1
|
|
return 0
|
|
|
|
parser.print_help()
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|