thaidinhz1 commited on
Commit
cc2a682
·
1 Parent(s): 67adee2

fix: strip whitespace from API keys

Browse files
Files changed (2) hide show
  1. src/query_rewriter.py +1 -1
  2. src/rag.py +1 -1
src/query_rewriter.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  from dotenv import load_dotenv
4
 
5
  load_dotenv()
6
- _client = Groq(api_key=os.environ["GROQ_API_KEY"])
7
 
8
  REWRITE_MODEL = "llama-3.3-70b-versatile"
9
 
 
3
  from dotenv import load_dotenv
4
 
5
  load_dotenv()
6
+ _client = Groq(api_key=os.environ["GROQ_API_KEY"].strip())
7
 
8
  REWRITE_MODEL = "llama-3.3-70b-versatile"
9
 
src/rag.py CHANGED
@@ -7,7 +7,7 @@ from src.reranker import rerank
7
  from src.query_rewriter import rewrite_query
8
 
9
  load_dotenv()
10
- client = Groq(api_key=os.environ["GROQ_API_KEY"])
11
 
12
  LLM_MODEL = "llama-3.3-70b-versatile"
13
 
 
7
  from src.query_rewriter import rewrite_query
8
 
9
  load_dotenv()
10
+ client = Groq(api_key=os.environ["GROQ_API_KEY"].strip())
11
 
12
  LLM_MODEL = "llama-3.3-70b-versatile"
13