41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
MODEL_CONFIG = {
|
|
"openai": {
|
|
"gpt-4o": {
|
|
"tool_calling_supported": True,
|
|
"processing_chunk_size": 10000,
|
|
"processing_chunk_overlap": 200,
|
|
"processing_min_chunk_size": 8000,
|
|
"processing_max_chunk_size": 12000,
|
|
"prompt_templates": [
|
|
"summary", "rag", "history", "encyclopedia",
|
|
"transcript", "html_parse", "pdf_parse"
|
|
]
|
|
},
|
|
"gpt-4o-mini": {
|
|
"tool_calling_supported": True,
|
|
"processing_chunk_size": 10000,
|
|
"processing_chunk_overlap": 200,
|
|
"processing_min_chunk_size": 8000,
|
|
"processing_max_chunk_size": 12000,
|
|
"prompt_templates": [
|
|
"summary", "rag", "history", "encyclopedia",
|
|
"transcript", "html_parse", "pdf_parse"
|
|
]
|
|
},
|
|
# Add other OpenAI models here
|
|
},
|
|
"anthropic": {
|
|
"claude-3-5-sonnet": {
|
|
"tool_calling_supported": True,
|
|
"processing_chunk_size": 10000,
|
|
"processing_chunk_overlap": 200,
|
|
"processing_min_chunk_size": 8000,
|
|
"processing_max_chunk_size": 12000,
|
|
"prompt_templates": [
|
|
"summary", "rag", "history", "encyclopedia",
|
|
"transcript", "html_parse", "pdf_parse"
|
|
]
|
|
},
|
|
# Add other Anthropic models here
|
|
},
|
|
} |