AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-04-21 13:39:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

support custom models

Browse Source
This commit is contained in:
rentianyue-jk 2024-07-16 10:22:54 +08:00
parent 8efd444ba2
commit 1b7ae5457f
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/api/src/lib/LLM-extraction/models.ts
View File

@ -48,7 +48,7 @@ function prepareOpenAIDoc(
export async function generateOpenAICompletions({ export async function generateOpenAICompletions({
client, client,
model = "gpt-4o", model = process.env.MODEL_NAME,
document, document,
schema, //TODO - add zod dynamic type checking schema, //TODO - add zod dynamic type checking
prompt = defaultPrompt, prompt = defaultPrompt,

2
docker-compose.yaml
View File

@ -12,6 +12,8 @@ x-common-service: &common-service
- PORT=${PORT:-3002} - PORT=${PORT:-3002}
- NUM_WORKERS_PER_QUEUE=${NUM_WORKERS_PER_QUEUE} - NUM_WORKERS_PER_QUEUE=${NUM_WORKERS_PER_QUEUE}
- OPENAI_API_KEY=${OPENAI_API_KEY} - OPENAI_API_KEY=${OPENAI_API_KEY}
- OPENAI_BASE_URL=${OPENAI_BASE_URL}
- MODEL_NAME=${MODEL_NAME:-gpt-4o}
- SLACK_WEBHOOK_URL=${SLACK_WEBHOOK_URL} - SLACK_WEBHOOK_URL=${SLACK_WEBHOOK_URL}
- SERPER_API_KEY=${SERPER_API_KEY} - SERPER_API_KEY=${SERPER_API_KEY}
- LLAMAPARSE_API_KEY=${LLAMAPARSE_API_KEY} - LLAMAPARSE_API_KEY=${LLAMAPARSE_API_KEY}