Update vector_store_retriever.py
Browse files
vector_store_retriever.py
CHANGED
@@ -169,8 +169,10 @@ class TextGeneratorTool(Tool):
|
|
169 |
inputs = ["prompt"]
|
170 |
outputs = ["generated_text"]
|
171 |
|
172 |
-
|
|
|
173 |
#self.retriever = db.as_retriever(search_kwargs={"k": 1})
|
|
|
174 |
|
175 |
def __call__(self, prompt: str):
|
176 |
result = predict(prompt, 0.9, 512, 0.6, 1.4)
|
|
|
169 |
inputs = ["prompt"]
|
170 |
outputs = ["generated_text"]
|
171 |
|
172 |
+
|
173 |
+
def __init__(self):
|
174 |
#self.retriever = db.as_retriever(search_kwargs={"k": 1})
|
175 |
+
pass # You might want to add some initialization logic here
|
176 |
|
177 |
def __call__(self, prompt: str):
|
178 |
result = predict(prompt, 0.9, 512, 0.6, 1.4)
|