diff --git a/src/deep-infra.ts b/src/deep-infra.ts index 41a037e..f0d41e2 100644 --- a/src/deep-infra.ts +++ b/src/deep-infra.ts @@ -33,12 +33,14 @@ export class DeepInfra implements Model { * @param model Model * @param system System prompt * @param prompt User prompt + * @param temperature Temperature */ constructor( private readonly token: string, private readonly model: string, private readonly system: Scalar, - private readonly prompt: Scalar + private readonly prompt: Scalar, + private readonly temperature: number ) { this.token = token; this.model = model; @@ -50,7 +52,7 @@ export class DeepInfra implements Model { method: 'POST', body: JSON.stringify({ model: this.model, - temperature: 0.5, + temperature: this.temperature, messages: [ { role: "system", @@ -59,9 +61,6 @@ export class DeepInfra implements Model { { role: "user", content: this.prompt.value() - // content: new UserPrompt( - // report - // ).value() } ], }), diff --git a/src/main.ts b/src/main.ts index 638b6d3..ee06980 100644 --- a/src/main.ts +++ b/src/main.ts @@ -163,7 +163,8 @@ async function run() { new QualityExpert(), new UserPrompt( new Titled(smart.title, body).asString() - ) + ), + 0.5 ).analyze(); await new Feedback( answer, diff --git a/src/pdd-model.ts b/src/pdd-model.ts index d19ff02..8d61230 100644 --- a/src/pdd-model.ts +++ b/src/pdd-model.ts @@ -60,7 +60,8 @@ export class PddModel implements Model { this.token, this.model, new TodoReviewer(), - this.prompt + this.prompt, + 0.7 ).analyze(); } return answer;