mirror of
https://github.com/prowler-cloud/prowler.git
synced 2026-07-23 12:31:54 +00:00
Use specific model snapshots
This commit is contained in:
@@ -1335,8 +1335,8 @@ class LighthouseConfig(RowLevelSecurityProtectedModel):
|
||||
|
||||
name = models.CharField(max_length=100, validators=[MinLengthValidator(3)])
|
||||
api_key = models.BinaryField(blank=False, null=False)
|
||||
model = models.CharField(max_length=50, default="gpt-4o")
|
||||
temperature = models.FloatField(default=0.7)
|
||||
model = models.CharField(max_length=50, default="gpt-4o-2024-08-06")
|
||||
temperature = models.FloatField(default=0)
|
||||
max_tokens = models.IntegerField(default=4000)
|
||||
business_context = models.TextField(
|
||||
blank=True,
|
||||
|
||||
@@ -80,22 +80,22 @@ ${
|
||||
|
||||
const initializeModels = async () => {
|
||||
const apiKey = await getAIKey();
|
||||
const aiConfig = await getAIConfiguration();
|
||||
const modelConfig = aiConfig?.data?.attributes;
|
||||
|
||||
// Initialize models without API keys
|
||||
const llm = new ChatOpenAI({
|
||||
model: "gpt-4o",
|
||||
temperature: 0,
|
||||
topP: 0,
|
||||
maxTokens: 5000,
|
||||
model: modelConfig?.model || "gpt-4",
|
||||
temperature: modelConfig?.temperature || 0,
|
||||
maxTokens: modelConfig?.max_tokens || 5000,
|
||||
apiKey: apiKey,
|
||||
tags: ["agent"],
|
||||
});
|
||||
|
||||
const supervisorllm = new ChatOpenAI({
|
||||
model: "gpt-4o",
|
||||
temperature: 0,
|
||||
topP: 0,
|
||||
maxTokens: 5000,
|
||||
model: modelConfig?.model || "gpt-4",
|
||||
temperature: modelConfig?.temperature || 0,
|
||||
maxTokens: modelConfig?.max_tokens || 5000,
|
||||
apiKey: apiKey,
|
||||
streaming: true,
|
||||
tags: ["supervisor"],
|
||||
|
||||
@@ -207,10 +207,13 @@ export default function ChatbotConfig() {
|
||||
size="md"
|
||||
isRequired
|
||||
>
|
||||
<SelectItem key="gpt-4o" value="gpt-4o">
|
||||
<SelectItem key="gpt-4o-2024-08-06" value="gpt-4o-2024-08-06">
|
||||
GPT-4o (Recommended)
|
||||
</SelectItem>
|
||||
<SelectItem key="gpt-4o-mini" value="gpt-4o-mini">
|
||||
<SelectItem
|
||||
key="gpt-4o-mini-2024-07-18"
|
||||
value="gpt-4o-mini-2024-07-18"
|
||||
>
|
||||
GPT-4o Mini
|
||||
</SelectItem>
|
||||
</Select>
|
||||
|
||||
Reference in New Issue
Block a user