Rulga commited on
Commit
b628df0
·
1 Parent(s): c562673

Refactor settings.py: Add configurations for GPT-3.5 Turbo and Claude-3-Haiku models, enhancing capabilities and documentation

Browse files
Files changed (1) hide show
  1. config/settings.py +71 -0
config/settings.py CHANGED
@@ -299,6 +299,77 @@ MODELS = {
299
  ],
300
  "documentation": "https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1"
301
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
302
  }
303
  }
304
 
 
299
  ],
300
  "documentation": "https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1"
301
  }
302
+ },
303
+ "gpt-3.5-turbo": {
304
+ "id": "gpt-3.5-turbo",
305
+ "name": "GPT-3.5 Turbo",
306
+ "description": "OpenAI's efficient and reliable chat model",
307
+ "type": "api",
308
+ "parameters": {
309
+ "max_length": 4096,
310
+ "temperature": 0.7,
311
+ "top_p": 0.9,
312
+ "repetition_penalty": 1.1,
313
+ },
314
+ "training": None, # API model, no local training
315
+ "details": {
316
+ "full_name": "GPT-3.5 Turbo",
317
+ "capabilities": [
318
+ "Strong multilingual support",
319
+ "Excellent instruction following",
320
+ "Reliable performance",
321
+ "Good understanding of legal concepts"
322
+ ],
323
+ "limitations": [
324
+ "API costs (but reasonable)",
325
+ "Rate limits apply",
326
+ "Knowledge cutoff date",
327
+ "Requires API key"
328
+ ],
329
+ "use_cases": [
330
+ "Legal consultation",
331
+ "Document analysis",
332
+ "Multilingual support",
333
+ "Complex reasoning tasks"
334
+ ],
335
+ "documentation": "https://platform.openai.com/docs/models/gpt-3-5"
336
+ }
337
+ },
338
+ "claude-3-haiku": {
339
+ "id": "claude-3-haiku-20240307",
340
+ "name": "Claude-3-Haiku",
341
+ "description": "Anthropic's fastest and most cost-effective Claude-3 model",
342
+ "type": "api",
343
+ "parameters": {
344
+ "max_length": 4096,
345
+ "temperature": 0.7,
346
+ "top_p": 0.9,
347
+ "repetition_penalty": 1.1,
348
+ },
349
+ "training": None, # API model, no local training
350
+ "details": {
351
+ "full_name": "Claude-3-Haiku",
352
+ "capabilities": [
353
+ "Excellent reasoning abilities",
354
+ "Strong multilingual support",
355
+ "High accuracy in legal domain",
356
+ "Fast response time",
357
+ "Cost-effective API usage"
358
+ ],
359
+ "limitations": [
360
+ "API costs (though reasonable)",
361
+ "Rate limits apply",
362
+ "Requires Anthropic API key",
363
+ "Shorter context window than Opus/Sonnet"
364
+ ],
365
+ "use_cases": [
366
+ "Legal document analysis",
367
+ "Quick legal consultations",
368
+ "Document drafting assistance",
369
+ "Research support"
370
+ ],
371
+ "documentation": "https://docs.anthropic.com/claude/docs/models-overview"
372
+ }
373
  }
374
  }
375