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(
|