Bima Ardhia commited on
Commit
85492ba
·
1 Parent(s): f6b9d15

add secur

Browse files
Files changed (4) hide show
  1. .gitignore +1 -0
  2. agent/retrive_agent.py +3 -2
  3. app/main.py +2 -3
  4. tools/retrive.py +3 -0
.gitignore CHANGED
@@ -1 +1,2 @@
1
  .vercel
 
 
1
  .vercel
2
+ .env
agent/retrive_agent.py CHANGED
@@ -1,7 +1,5 @@
1
- from dotenv import load_dotenv
2
  from typing import Any, Dict, List
3
 
4
- load_dotenv()
5
  from langchain_openai import ChatOpenAI
6
  from langchain.prompts.prompt import PromptTemplate
7
  from langchain_core.tools import Tool
@@ -11,6 +9,9 @@ from langchain.agents import (
11
  )
12
  from langchain import hub
13
  from tools.retrive import retrieve_berita, retrieve_umkm, retrieve_wisata, get_current_time, retrieve_wikipedia_info
 
 
 
14
 
15
 
16
  def run_llm(query: str, chat_history: List[Dict[str, Any]] = []) -> str:
 
 
1
  from typing import Any, Dict, List
2
 
 
3
  from langchain_openai import ChatOpenAI
4
  from langchain.prompts.prompt import PromptTemplate
5
  from langchain_core.tools import Tool
 
9
  )
10
  from langchain import hub
11
  from tools.retrive import retrieve_berita, retrieve_umkm, retrieve_wisata, get_current_time, retrieve_wikipedia_info
12
+ import os
13
+
14
+ OPENAI_API_KEY= os.getenv("OPENAI_API_KEY")
15
 
16
 
17
  def run_llm(query: str, chat_history: List[Dict[str, Any]] = []) -> str:
app/main.py CHANGED
@@ -3,14 +3,13 @@ from firebase_admin import credentials, firestore
3
  from langchain_google_firestore import FirestoreChatMessageHistory
4
  from langchain.memory import ConversationBufferWindowMemory
5
  from langchain.schema import HumanMessage, AIMessage
6
- from dotenv import load_dotenv
7
  import firebase_admin
8
  import uuid
9
  from agent.retrive_agent import run_llm
 
10
 
11
 
12
-
13
- load_dotenv()
14
 
15
 
16
  PROJECT_ID = "recommendation-system-mage"
 
3
  from langchain_google_firestore import FirestoreChatMessageHistory
4
  from langchain.memory import ConversationBufferWindowMemory
5
  from langchain.schema import HumanMessage, AIMessage
 
6
  import firebase_admin
7
  import uuid
8
  from agent.retrive_agent import run_llm
9
+ import os
10
 
11
 
12
+ OPENAI_API_KEY= os.getenv("OPENAI_API_KEY")
 
13
 
14
 
15
  PROJECT_ID = "recommendation-system-mage"
tools/retrive.py CHANGED
@@ -6,6 +6,9 @@ import json
6
  from datetime import datetime
7
  from langchain_community.tools import WikipediaQueryRun
8
  from langchain_community.utilities import WikipediaAPIWrapper
 
 
 
9
 
10
  # Initialize Pinecone client
11
  pc = Pinecone(api_key="pcsk_6vVrG6_2ZUYuJwQcrKefGPodrfDuYAXB4AKD2ZENoXF7qmAahuSAYWijPTaBhkkb8DoB4M")
 
6
  from datetime import datetime
7
  from langchain_community.tools import WikipediaQueryRun
8
  from langchain_community.utilities import WikipediaAPIWrapper
9
+ import os
10
+ OPENAI_API_KEY= os.getenv("OPENAI_API_KEY")
11
+
12
 
13
  # Initialize Pinecone client
14
  pc = Pinecone(api_key="pcsk_6vVrG6_2ZUYuJwQcrKefGPodrfDuYAXB4AKD2ZENoXF7qmAahuSAYWijPTaBhkkb8DoB4M")