inoid commited on
Commit
be4bfb2
1 Parent(s): 6eed986

Fix bugs about positional arguments in GeminiLLM/getLLM function.

Browse files
Files changed (1) hide show
  1. llm_call.py +1 -1
llm_call.py CHANGED
@@ -29,7 +29,7 @@ class GeminiLLM():
29
 
30
  return (retriever, vectorstore)
31
 
32
- def getLLM(self, documents ):
33
  if os.getenv('GOOGLE_GEMINI_TOKEN') is None:
34
  raise ValueError("GOOGLE_GEMINI_TOKEN environment variable not set")
35
  else:
 
29
 
30
  return (retriever, vectorstore)
31
 
32
+ def getLLM(self ):
33
  if os.getenv('GOOGLE_GEMINI_TOKEN') is None:
34
  raise ValueError("GOOGLE_GEMINI_TOKEN environment variable not set")
35
  else: