Skip to content

Commit 8ce41f1

Browse files
committed
patch(provider): add the temp and max tokens config values for users to configured on a per-provider basis
1 parent ae35f97 commit 8ce41f1

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

src/main/java/com/redhat/podmortem/common/model/provider/AIProviderConfig.java

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ public class AIProviderConfig implements Serializable {
2020
private int maxRetries = 3;
2121
private boolean cachingEnabled = true;
2222
private String promptTemplate;
23+
private int maxTokens = 500;
24+
private double temperature = 0.3;
2325

2426
public AIProviderConfig() {}
2527

@@ -95,6 +97,22 @@ public void setPromptTemplate(String promptTemplate) {
9597
this.promptTemplate = promptTemplate;
9698
}
9799

100+
public int getMaxTokens() {
101+
return maxTokens;
102+
}
103+
104+
public void setMaxTokens(int maxTokens) {
105+
this.maxTokens = maxTokens;
106+
}
107+
108+
public double getTemperature() {
109+
return temperature;
110+
}
111+
112+
public void setTemperature(double temperature) {
113+
this.temperature = temperature;
114+
}
115+
98116
@Override
99117
public boolean equals(Object o) {
100118
if (this == o) return true;

0 commit comments

Comments
 (0)