John Graham Reynolds commited on
Commit
e49ec86
·
1 Parent(s): 4dd1424

add missing return statement

Browse files
Files changed (1) hide show
  1. chain.py +4 -3
chain.py CHANGED
@@ -45,6 +45,7 @@ class ChainBuilder:
45
  def load_and_cache_embedding_model(model_name):
46
  embeddings = HuggingFaceEmbeddings(model_name=model_name, cache_folder="./langchain_cache/") # this cache isnt working because were in the Docker container
47
  # update this to read from a presaved cache of bge-large
 
48
 
49
  return load_and_cache_embedding_model(model_name)
50
 
@@ -66,7 +67,7 @@ class ChainBuilder:
66
  columns=["name", "description"],
67
  ).as_retriever(search_kwargs=search_kwargs)
68
 
69
- return vector_search_as_retriever
70
 
71
  return get_and_cache_retriever(endpoint, index_name, embeddings, search_kwargs)
72
 
@@ -104,7 +105,7 @@ class ChainBuilder:
104
  ("user", "{question}"),
105
  ]
106
  )
107
- return prompt
108
 
109
  # Format the converastion history to fit into the prompt template above.
110
  # **** TODO after only a few statements this will likely overflow the context window
@@ -145,7 +146,7 @@ class ChainBuilder:
145
  endpoint=endpoint,
146
  extra_params=extra_params,
147
  )
148
- return model
149
 
150
  return get_and_cache_model(endpoint, extra_params)
151
 
 
45
  def load_and_cache_embedding_model(model_name):
46
  embeddings = HuggingFaceEmbeddings(model_name=model_name, cache_folder="./langchain_cache/") # this cache isnt working because were in the Docker container
47
  # update this to read from a presaved cache of bge-large
48
+ return embeddings # return directly?
49
 
50
  return load_and_cache_embedding_model(model_name)
51
 
 
67
  columns=["name", "description"],
68
  ).as_retriever(search_kwargs=search_kwargs)
69
 
70
+ return vector_search_as_retriever # return directly?
71
 
72
  return get_and_cache_retriever(endpoint, index_name, embeddings, search_kwargs)
73
 
 
105
  ("user", "{question}"),
106
  ]
107
  )
108
+ return prompt # return directly?
109
 
110
  # Format the converastion history to fit into the prompt template above.
111
  # **** TODO after only a few statements this will likely overflow the context window
 
146
  endpoint=endpoint,
147
  extra_params=extra_params,
148
  )
149
+ return model # return directly?
150
 
151
  return get_and_cache_model(endpoint, extra_params)
152