chore: fixed frontend call

ubiquity-os-marketplace · Dec 19, 2024 · 66c1e77 · 66c1e77
1 parent 85029f0
commit 66c1e77
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/parser/content-evaluator-module.ts b/src/parser/content-evaluator-module.ts
@@ -193,6 +193,7 @@ export class ContentEvaluatorModule extends BaseModule {
       const maxTokens = this._calculateMaxTokens(dummyResponse);
 
       const promptForComments = this._generatePromptForComments(specification, comments, allComments);
+      // check token count
       commentRelevances = await this._submitPrompt(promptForComments, maxTokens);
     }
 
@@ -201,6 +202,7 @@ export class ContentEvaluatorModule extends BaseModule {
       const maxTokens = this._calculateMaxTokens(dummyResponse);
 
       const promptForPrComments = this._generatePromptForPrComments(specification, prComments);
+      // check token count
       prCommentRelevances = await this._submitPrompt(promptForPrComments, maxTokens);
     }
 

diff --git a/src/web/api/index.ts b/src/web/api/index.ts
@@ -37,7 +37,7 @@ const app = {
     if (
       request.method === "POST" &&
       new URL(request.url).pathname === "/" &&
-      request.headers.get("origin") === "http://localhost:4000"
+      request.headers.get("referer")?.includes("http://localhost:4000")
     ) {
       try {
         const originalBody = await request.json();