heymenn commited on
Commit
2783254
·
verified ·
1 Parent(s): 7763377

Update src/processor.py

Browse files
Files changed (1) hide show
  1. src/processor.py +3 -2
src/processor.py CHANGED
@@ -1,7 +1,8 @@
1
  from fuzzywuzzy import fuzz
2
  from google.genai import Client, types
3
- import json
4
  from datasets import load_dataset
 
 
5
 
6
 
7
  def search_and_retrieve(user_input, config):
@@ -79,7 +80,7 @@ def generate_tech(user_input, user_instructions):
79
  </USER_INPUT>
80
  """
81
 
82
- client = Client(api_key="AIzaSyCHcw26RRXZAlb1twhc_cRkoqCFW3e8QKk")
83
 
84
  # Define the grounding tool
85
  grounding_tool = types.Tool(
 
1
  from fuzzywuzzy import fuzz
2
  from google.genai import Client, types
 
3
  from datasets import load_dataset
4
+ import json
5
+ import os
6
 
7
 
8
  def search_and_retrieve(user_input, config):
 
80
  </USER_INPUT>
81
  """
82
 
83
+ client = Client(api_key=os.getenv("GEMINI_API_KEY"))
84
 
85
  # Define the grounding tool
86
  grounding_tool = types.Tool(