Update src/processor.py
Browse files- 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="
|
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(
|