Fix Pinecone import (#18)

src/chat.py

@@ -77,7 +77,6 @@ if __name__ == "__main__":
     parser.add_argument("--llm_provider", default="anthropic", choices=["openai", "anthropic", "ollama"])
     parser.add_argument(
         "--llm_model",
-        default="claude-3-opus-20240229",
         help="The LLM name. Must be supported by the provider specified via --llm_provider.",
     )
     parser.add_argument("--vector_store_type", default="pinecone", choices=["pinecone", "marqo"])
@@ -94,6 +93,14 @@ if __name__ == "__main__":
     )
     args = parser.parse_args()
 
+    if not args.llm_model:
+        if args.llm_provider == "openai":
+            args.llm_model = "gpt-4"
+        elif args.llm_provider == "anthropic":
+            args.llm_model = "claude-3-opus-20240229"
+        else:
+            raise ValueError("Please specify --llm_model")
+
     rag_chain = build_rag_chain(args)
 
     def _predict(message, history):

src/vector_store.py

@@ -4,7 +4,7 @@ from abc import ABC, abstractmethod
 from typing import Dict, Generator, List, Tuple
 
 import marqo
-from langchain_community.vectorstores import Marqo
+from langchain_community.vectorstores import Marqo, Pinecone as LangChainPinecone
 from langchain_core.documents import Document
 from langchain_openai import OpenAIEmbeddings
 from pinecone import Pinecone
@@ -61,7 +61,7 @@ class PineconeVectorStore(VectorStore):
         self.index.upsert(vectors=pinecone_vectors, namespace=self.namespace)
 
     def to_langchain(self):
-        return Pinecone.from_existing_index(
+        return LangChainPinecone.from_existing_index(
             index_name=self.index_name, embedding=OpenAIEmbeddings(), namespace=self.namespace
         )