juliaturc commited on
Commit
dfb1df2
·
1 Parent(s): 4c99f56

Relax gradio version (causes urllib3 conflict in certain setups) (#20)

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. src/vector_store.py +2 -1
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  GitPython==3.1.43
2
  Pygments==2.18.0
3
- gradio==4.42.0
4
  langchain==0.2.14
5
  langchain-community==0.2.12
6
  langchain-anthropic==0.1.23
 
1
  GitPython==3.1.43
2
  Pygments==2.18.0
3
+ gradio>=4.26.0
4
  langchain==0.2.14
5
  langchain-community==0.2.12
6
  langchain-anthropic==0.1.23
src/vector_store.py CHANGED
@@ -99,7 +99,8 @@ class MarqoVectorStore(VectorStore):
99
  def build_from_args(args: dict) -> VectorStore:
100
  """Builds a vector store from the given command-line arguments."""
101
  if args.vector_store_type == "pinecone":
102
- return PineconeVectorStore(index_name=args.index_name, namespace=args.repo_id, dimension=args.embedding_size)
 
103
  elif args.vector_store_type == "marqo":
104
  return MarqoVectorStore(url=args.marqo_url, index_name=args.index_name)
105
  else:
 
99
  def build_from_args(args: dict) -> VectorStore:
100
  """Builds a vector store from the given command-line arguments."""
101
  if args.vector_store_type == "pinecone":
102
+ dimension = args.embedding_size if "embedding_size" in args else None
103
+ return PineconeVectorStore(index_name=args.index_name, namespace=args.repo_id, dimension=dimension)
104
  elif args.vector_store_type == "marqo":
105
  return MarqoVectorStore(url=args.marqo_url, index_name=args.index_name)
106
  else: