Skip to content

Commit aa24325

Browse files
author
xin.xu1
committed
fix ai config
1 parent 2d0e393 commit aa24325

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ app/mrmax.py
44
db/*
55
!db/.gitkeep
66
/venv/
7+
.config.py
78
__pycache__/
89
.DS_Store
910

config/config.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
11
# api 接口封装类
2+
23
llm_api_impl = "large_model.api.default_api.DefaultApi"
34

45
# DeepSeek配置示例
56
# api 配置方式参考 docs/config.md
67
# 默认使用认UnionLLM,参考:https://github.com/EvalsOne/UnionLLM/tree/main/docs
78
# UnionLLM兼容LiteLLM,参考LiteLLM文档:https://docs.litellm.ai/docs
9+
810
api_config = {
911
"api_key": "your deepseek key",
1012
"model": 'deepseek-chat',
@@ -26,6 +28,14 @@
2628
# "provider": "ollama",
2729
# }
2830

31+
# demo-azure
32+
# api_config = {
33+
# "AZURE_API_KEY": "*",
34+
# "AZURE_API_BASE": "https://*.openai.azure.com",
35+
# "AZURE_API_VERSION": "2024-10-21",
36+
# "model": "azure/o1-mini",
37+
# }
38+
2939
# Prompt
3040
GPT_MESSAGE = """
3141
你是一位资深编程专家,gitlab的分支代码变更将以git diff 字符串的形式提供,请你帮忙review本段代码。然后你review内容的返回内容必须严格遵守下面的格式,包括标题内容。模板中的变量内容解释:

0 commit comments

Comments
 (0)