cleanup

.gitlab-ci.yml

@@ -22,21 +22,13 @@ stages:
   image:
     name: gcr.io/kaniko-project/executor:debug
     entrypoint: [""]
-  variables:
-    IMAGE_TAG: latest
-    GIT_SUBMODULE_STRATEGY: recursive
-    GIT_SUBMODULE_DEPTH: 1
-    GIT_SUBMODULE_UPDATE_FLAGS: --jobs 4
   before_script:
     - echo "{\"auths\":{\"${CI_REGISTRY}\":{\"auth\":\"$(printf "%s:%s" "${CI_REGISTRY_USER}" "${CI_REGISTRY_PASSWORD}" | base64 | tr -d '\n')\"},\"$CI_DEPENDENCY_PROXY_SERVER\":{\"auth\":\"$(printf "%s:%s" ${CI_DEPENDENCY_PROXY_USER} "${CI_DEPENDENCY_PROXY_PASSWORD}" | base64 | tr -d '\n')\"}}}" > /kaniko/.docker/config.json
-
-build:
-  extends: .build
   script:
     - /kaniko/executor
       --context "${CI_PROJECT_DIR}"
       --dockerfile "${CI_PROJECT_DIR}/Dockerfile"
-      --destination "${CI_REGISTRY_IMAGE}/jupyter:${DOCKER_TAG}"
+      --destination "${CI_REGISTRY_IMAGE}:${DOCKER_TAG}"
   rules:
     - changes:
       - ${CI_PROJECT_DIR}/Dockerfile

api_writer/fast_app.py

-import gradio as gr
-from fastapi import FastAPI
-from api_writer.test_ui import demo
-
-
-api = FastAPI()
-
-
-@api.get("/")
-def read_main():
-    return {"message": "This is your main app"}
-
-
-app = gr.mount_gradio_app(api, demo, path="/staging/llm-tutorial")
+# import gradio as gr
+# from fastapi import FastAPI
+# from api_writer.test_ui import demo
+#
+#
+# api = FastAPI()
+#
+#
+# @api.get("/")
+# def read_main():
+#     return {"message": "This is your main app"}
+#
+#
+# app = gr.mount_gradio_app(api, demo, path="/staging/llm-tutorial")

api_writer/test_ui.py

@@ -6,16 +6,14 @@ import pandas as pd
 from copy import deepcopy
 from opik import track, opik_context
 from api_writer.nomad_api import make_query
-from api_writer.utils import add_values, check_keys, load_query, load_config
+from api_writer.utils import check_keys, load_query, load_config
+
 
 if os.getenv["USE_VLLM_API"]:
     from api_writer.vllm_api import http_bot
 else:
     from api_writer.vllm_model import http_bot
 
-search_link_url = "https://nomad-lab.eu/prod/v1/gui/search/entries/entry/id/"
-print("Caching context...")
-http_bot("warmup")
 
 # try:
 #     os.environ["OPIK_API_KEY"]
@@ -24,22 +22,13 @@ http_bot("warmup")
 
 os.environ["OPIK_WORKSPACE"] = "pilar12"
 
+n = 2
+search_link_url = "https://nomad-lab.eu/prod/v1/gui/search/entries/entry/id/"
 project_name = os.getenv("OPIK_PROJECT_NAME", "NOMAD_Demo_test_ui")
-
 sampling_params = load_config("api_writer/sampling_params_2.yaml")
 
-n = 2
-
-
-def format_num_response(response):
-    response = response[response.index("JSON") + 5 :]
-    values = response[response.index("'''") + 3 :].strip().split("\n")
-    base_json = response[: response.index("'''")].strip()
-    if values != [""]:
-        response = add_values(base_json, values)
-    else:
-        return base_json
-    return response
+print("Caching context...")
+http_bot("warmup")
 
 
 def process_api_output(api_output: dict) -> pd.DataFrame: