feat(#88): model ref

tracehubpm · Jun 7, 2024 · e600dca · e600dca
1 parent 6d61c61
commit e600dca
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/pom.xml b/pom.xml
@@ -103,7 +103,7 @@ SOFTWARE.
     <dependency>
       <groupId>com.jcabi</groupId>
       <artifactId>jcabi-github</artifactId>
-       <version>${jcabi-github.version}</version>
+      <version>${jcabi-github.version}</version>
     </dependency>
     <dependency>
       <groupId>org.hamcrest</groupId>

diff --git a/src/main/java/git/tracehub/codereview/action/AnalysisRoutine.java b/src/main/java/git/tracehub/codereview/action/AnalysisRoutine.java
@@ -95,7 +95,7 @@ public void exec(final Pull pull, final String model) throws Exception {
             credential = System.getProperty(platf);
         }
         final Model analysis = new AutoModel(
-            credential, this.platform, system, prompt
+            credential, this.platform, system, prompt, model
         ).value();
         final String score = analysis.completion();
         if (score.toLowerCase(Locale.ROOT).contains("excellent")) {
@@ -127,7 +127,8 @@ public void exec(final Pull pull, final String model) throws Exception {
                                             new PullChanges(pull)
                                         )
                                     )
-                                ).asString()
+                                ).asString(),
+                                model
                             ).value().completion(),
                             model
                         ),

diff --git a/src/main/java/git/tracehub/codereview/action/AutoModel.java b/src/main/java/git/tracehub/codereview/action/AutoModel.java
@@ -60,6 +60,11 @@ public final class AutoModel implements Scalar<Model> {
      */
     private final String prompt;
 
+    /**
+     * Model ref.
+     */
+    private final String ref;
+
     @Override
     public Model value() throws Exception {
         final String platf = this.platform.value();
@@ -70,7 +75,7 @@ public Model value() throws Exception {
                     this.credential,
                     new CompleteJson(
                         new Simple(
-                            System.getenv().get("INPUT_DEEPINFRA_MODEL"),
+                            this.ref,
                             0.7,
                             512
                         ),
@@ -85,7 +90,7 @@ public Model value() throws Exception {
                     this.credential,
                     new OpenJson(
                         new Simple(
-                            System.getenv().get("INPUT_OPENAI_MODEL"),
+                            this.ref,
                             0.7,
                             512
                         ),