eigent/backend/app/model/model_platform.py
puzhen 63b723e1cf merge main into feat/ask_with_single_agent
Resolve conflicts in:
- server/uv.lock (revision number, greenlet wheel format)
- src/store/chatStore.ts (merge target/displayContent and executionId/projectId params)
- src/components/ChatBox/BottomBox/InputBox.tsx (merge mention and trigger/expanded features)
- src/components/ChatBox/index.tsx (merge mention/direct-agent features with new unified layout)
2026-03-03 11:45:42 +00:00

44 lines
1.6 KiB
Python

# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
from typing import Annotated, Final
from pydantic import BeforeValidator
PLATFORM_ALIAS_MAPPING: Final[dict[str, str]] = {
"z.ai": "openai-compatible-model",
"ModelArk": "openai-compatible-model",
"grok": "openai-compatible-model",
"llama.cpp": "openai-compatible-model",
}
def normalize_model_platform(platform: str) -> str:
"""Normalize provider aliases to supported model platform names."""
return PLATFORM_ALIAS_MAPPING.get(platform, platform)
def normalize_optional_model_platform(platform: str | None) -> str | None:
"""Optional variant of normalize_model_platform."""
if platform is None:
return None
return normalize_model_platform(platform)
NormalizedModelPlatform = Annotated[
str, BeforeValidator(normalize_model_platform)
]
NormalizedOptionalModelPlatform = Annotated[
str | None, BeforeValidator(normalize_optional_model_platform)
]