添加RL方面的功能，同时去掉了redis依赖 (#1776)

* added RL functions

* removed redis

Author: konpoku

README.md

@@ -23,12 +23,6 @@ EESAST 后端 API
 - node 16 / npm
 - yarn
 - TypeScript
-- Redis
-
-### Before Dev
-
-需要在测试端安装和启动Redis
-具体见[Redis安装文档](https://redis.io/docs/latest/operate/oss_and_stack/install/archive/install-redis/)
 
 ### 工具
 

package.json

@@ -28,7 +28,6 @@
     "graphql": "16.14.0",
     "graphql-request": "6.1.0",
     "html-to-text": "10.0.0",
-    "ioredis": "5.10.1",
     "isemail": "3.2.0",
     "js-yaml": "4.1.1",
     "jsonwebtoken": "9.0.2",

src/hasura/contest.ts

@@ -1582,6 +1582,55 @@ export const get_team_software_final_submission_count: any = async (
   );
 };
 
+export const add_team_RL_code: any = async (
+  team_id: string,
+  code_url: string,
+) => {
+  const insert_team_RL_code: any = await client.request(
+    gql`
+      mutation add_team_RL_code($team_id: uuid!, $url: String!) {
+        insert_contest_team_RL_code_one(
+          object: { team_id: $team_id, url: $url }
+        ) {
+          submit_id
+          team_id
+          created_at
+          url
+        }
+      }
+    `,
+    {
+      team_id: team_id,
+      url: code_url,
+    },
+  );
+
+  return insert_team_RL_code.insert_contest_team_RL_code_one;
+};
+
+export const get_team_RL_code_submissions: any = async (team_id: string) => {
+  const query_team_RL_code_submissions: any = await client.request(
+    gql`
+      query get_team_RL_code_submissions($team_id: uuid!) {
+        contest_team_RL_code(
+          where: { team_id: { _eq: $team_id } }
+          order_by: { created_at: asc }
+        ) {
+          submit_id
+          team_id
+          created_at
+          url
+        }
+      }
+    `,
+    {
+      team_id: team_id,
+    },
+  );
+
+  return query_team_RL_code_submissions.contest_team_RL_code;
+};
+
 /**
  *
  * @param {uuid} contest_id         ID
@@ -2713,6 +2762,37 @@ export const get_software_contest_deadline: any = async (
   );
 };
 
+export const set_software_contest_deadline: any = async (
+  contest_id: string,
+  deadline: string,
+) => {
+  const upsert_software_contest_deadline: any = await client.request(
+    gql`
+      mutation set_software_contest_deadline(
+        $contest_id: uuid!
+        $deadline: timestamptz!
+      ) {
+        insert_contest_soft_code_deadline_one(
+          object: { contest_id: $contest_id, deadline: $deadline }
+          on_conflict: {
+            constraint: contest_soft_code_deadline_pkey
+            update_columns: [deadline]
+          }
+        ) {
+          contest_id
+          deadline
+        }
+      }
+    `,
+    {
+      contest_id: contest_id,
+      deadline: deadline,
+    },
+  );
+
+  return upsert_software_contest_deadline.insert_contest_soft_code_deadline_one;
+};
+
 export const get_contest_id_from_team_id: any = async (team_id: string) => {
   const query_contest_id_from_team_id: any = await client.request(
     gql`
@@ -2728,3 +2808,58 @@ export const get_contest_id_from_team_id: any = async (team_id: string) => {
   );
   return query_contest_id_from_team_id.contest_team[0]?.contest_id ?? undefined;
 };
+
+/**
+ * Get all RL scores for a contest, ordered by score descending.
+ * @param {string} contest_id
+ * @returns {Array<{team_id, score, team_name}>}
+ */
+export const get_all_RL_scores: any = async (contest_id: string) => {
+  const result: any = await client.request(
+    gql`
+      query GetAllRLScores($contest_id: uuid!) {
+        contest_team_RL_score(
+          where: { contest_team: { contest_id: { _eq: $contest_id } } }
+          order_by: { score: desc_nulls_last }
+        ) {
+          team_id
+          score
+          contest_team {
+            team_name
+          }
+        }
+      }
+    `,
+    { contest_id },
+  );
+  return result.contest_team_RL_score ?? [];
+};
+
+/**
+ * Upsert (insert or update) a team's RL score.
+ * @param {string} team_id
+ * @param {number} score
+ * @returns {string} team_id
+ */
+export const upsert_team_RL_score: any = async (
+  team_id: string,
+  score: number,
+) => {
+  const result: any = await client.request(
+    gql`
+      mutation UpsertTeamRLScore($team_id: uuid!, $score: Int!) {
+        insert_contest_team_RL_score_one(
+          object: { team_id: $team_id, score: $score }
+          on_conflict: {
+            constraint: contest_team_RL_score_pkey
+            update_columns: [score]
+          }
+        ) {
+          team_id
+        }
+      }
+    `,
+    { team_id, score },
+  );
+  return result.insert_contest_team_RL_score_one?.team_id ?? undefined;
+};

src/hasura/llm.ts

@@ -51,6 +51,38 @@ export const set_llm_usage_by_uuid = async (
   return query.update_llm_usage_by_pk;
 };
 
+export const update_llm_usage_by_uuid = async (
+  uuid: string,
+  tokens_to_add: number,
+) => {
+  const query: any = await client.request(
+    gql`
+      mutation UpdateLlmUsageByUuid(
+        $uuid: uuid!
+        $tokens_to_add: bigint!
+        $updated_at: timestamptz!
+      ) {
+        update_llm_usage_by_pk(
+          pk_columns: { uuid: $uuid }
+          _inc: { total_tokens_used: $tokens_to_add }
+          _set: { last_updated_at: $updated_at }
+        ) {
+          uuid
+          total_tokens_used
+          token_limit
+          last_updated_at
+        }
+      }
+    `,
+    {
+      uuid,
+      tokens_to_add,
+      updated_at: new Date().toISOString(),
+    },
+  );
+  return query.update_llm_usage_by_pk;
+};
+
 export const get_user_llm_usage = async (student_no: string) => {
   const query: any = await client.request(
     gql`

src/helpers/llm_cron.ts

@@ -4,7 +4,6 @@ import http from "http";
 import app from "./app";
 import { GraphQLClient } from "graphql-request";
 import { queue_element } from "./helpers/docker_queue";
-import { llm_cron } from "./helpers/llm_cron";
 // import docker_cron from "./helpers/docker_queue";
 
 const debug = Debug("eesast-api");
@@ -36,8 +35,6 @@ export const docker_queue: queue_element[] = [];
 // weekly_cron();
 // weekly_init();
 
-llm_cron();
-
 const port = normalizePort(process.env.PORT || "28888");
 app.set("port", port);