|
@@ -13,6 +13,7 @@ from sqlalchemy.orm import sessionmaker
|
|
|
from pqai_agent import configs
|
|
|
|
|
|
from pqai_agent import chat_service, prompt_templates
|
|
|
+from pqai_agent.chat_service import OpenAICompatible
|
|
|
from pqai_agent.logging import logger, setup_root_logger
|
|
|
from pqai_agent.toolkit import global_tool_map
|
|
|
from pqai_agent.data_models.agent_configuration import AgentConfiguration
|
|
@@ -106,7 +107,7 @@ def list_models():
|
|
|
{
|
|
|
'model_type': 'openai_compatible',
|
|
|
'model_name': model_name,
|
|
|
- 'display_name': model_display_name
|
|
|
+ 'display_name': f"{model_display_name} ({OpenAICompatible.get_price(model_name).get_cny_brief()})"
|
|
|
}
|
|
|
for model_display_name, model_name in models.items()
|
|
|
]
|