Chris4K commited on
Commit
8ea7ad4
1 Parent(s): 3cf17c6

Update vector_store_retriever.py

Browse files
Files changed (1) hide show
  1. vector_store_retriever.py +3 -1
vector_store_retriever.py CHANGED
@@ -169,8 +169,10 @@ class TextGeneratorTool(Tool):
169
  inputs = ["prompt"]
170
  outputs = ["generated_text"]
171
 
172
- def __init__(self):
 
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)