Merge pull request #36 from Knox-AAU/concept-linking

Fixed some bugs in docker
Knox-AAU · Dec 15, 2023 · 03f0934 · 03f0934
2 parents cc1894e + d276902
commit 03f0934
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/concept_linking/solutions/PromptEngineering/main.py b/concept_linking/solutions/PromptEngineering/main.py
@@ -8,13 +8,16 @@
 from relation_extraction.knowledge_graph_messenger import KnowledgeGraphMessenger
 from concept_linking.tools.triple_helper import *
 
-# Local API url
+# Local API url python
 api_url = "http://127.0.0.1:5000/llama"
 
+# Local API url docker
+# api_url = "http://llama-cpu-server:5000/llama"
+
 # Remote API url
 # api_url = "http://knox-proxy01.srv.aau.dk/llama-api/llama"
 
-headers = {"Content-Type": "application/json"}
+headers = {"Access-Authorization": os.getenv("ACCESS_SECRET"), "Content-Type": "application/json"}
 
 PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../"))
 
@@ -181,7 +184,7 @@ def perform_entity_type_classification(post_json, output_file_path=None, output_
 
 
 if __name__ == '__main__':
-    input_file = os.path.join(PROJECT_ROOT, "data/files/EvaluationData/evaluationSet_EN.json")
+    input_file = os.path.join(PROJECT_ROOT, "data/files/EvaluationData/evaluationSet_EN_small.json")
     output_file = os.path.join(PROJECT_ROOT, "data/files/PromptEngineering/output.json")
 
     f = open(input_file,  encoding="utf-8")

diff --git a/concept_linking/tools/LlamaServer/docker-compose.yml b/concept_linking/tools/LlamaServer/docker-compose.yml
@@ -6,4 +6,4 @@ services:
     volumes:
       - ./concept_linking/tools/LlamaServer/llama-2-7b-chat.Q2_K.gguf:/app/concept_linking/tools/LlamaServer/llama-2-7b-chat.Q2_K.gguf
     ports:
-      - "5000:5000"
+      - "5000:5000"