Updates

.github/workflows/CICD_Huggingface.yml

@@ -14,6 +14,16 @@ jobs:
         with:
           fetch-depth: 0
           lfs: true
+
+      - name: Create .env file to store API key
+        run: |
+          echo "GEMINI_API_KEY=${{ secrets.GEMINI_API_KEY }}" >> ./.env
+          echo "NEO4J_URI=${{ secrets.NEO4J_URI }}" >> ./.env
+          echo "NEO4J_USERNAME=${{ secrets.NEO4J_USERNAME }}" >> ./.env
+          echo "NEO4J_PASSWORD=${{ secrets.NEO4J_PASSWORD }}" >> ./.env
+          echo "TAVILY_API_KEY=${{ secrets.TAVILY_API_KEY }}" >> ./.env
+
+
       - name: Push to hub
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}

utils.py

@@ -35,8 +35,8 @@ def init_():
 def get_llm_response(query):
     # Connect to Neo4J Knowledge Graph
     knowledge_graph, llm_chat = init_()
-    cypher_prompt = load_prompt("prompts/cypher_prompt.yaml")
-    qa_prompt = load_prompt("prompts/qa_prompt.yaml")
+    cypher_prompt = load_prompt("Agent/prompts/cypher_prompt.yaml")
+    qa_prompt = load_prompt("Agent/prompts/qa_prompt.yaml")
 
     CYPHER_GENERATION_PROMPT = PromptTemplate(**cypher_prompt)
     QA_GENERATION_PROMPT = PromptTemplate(**qa_prompt)