init

requirements.txt

@@ -1,3 +1,8 @@
--e .
-pandas==1.5.3
-numpy==1.24.2
+gcsfs
+s3fs
+adlfs
+universal_pathlib
+chromadb
+langchain
+langchainhub
+gpt4all

sandbox_alexis/main.py

@@ -0,0 +1,18 @@
+from langchain.vectorstores.chroma import Chroma
+from storage_backend import get_storage_root_path, StorageBackend
+from langchain.embeddings import GPT4AllEmbeddings
+
+from text_splitter import load_and_split_document
+
+data = """One of the most important things I didn't understand about the world when I was a child is the degree to which the returns for performance are superlinear.
+
+Teachers and coaches implicitly told us the returns were linear. "You get out," I heard a thousand times, "what you put in." They meant well, but this is rarely true. If your product is only half as good as your competitor's, you don't get half as many customers. You get no customers, and you go out of business.It's obviously true that the returns for performance are superlinear in business. Some think this is a flaw of capitalism, and that if we changed the rules it would stop being true. But superlinear returns for performance are a feature of the world, not an artifact of rules we've invented. We see the same pattern in fame, power, military victories, knowledge, and even benefit to humanity. In all of these, the rich get richer. [1]You can't understand the world without understanding the concept of superlinear returns. And if you're ambitious you definitely should, because this will be the wave you surf on.
+
+It may seem as if there are a lot of different situations with superlinear returns, but as far as I can tell they reduce to two fundamental causes: exponential growth and thresholds.The most obvious case of superlinear returns is when you're working on something that grows exponentially. For example, growing bacterial cultures. When they grow at all, they grow exponentially. But they're tricky to grow. Which means the difference in outcome between someone who's adept at it and someone who's not is very great.Startups can also grow exponentially, and we see the same pattern there. Some manage to achieve high growth rates. Most don't. And as a result you get qualitatively different outcomes: the companies with high growth rates tend to become immensely valuable, while the ones with lower growth rates may not even survive.Y Combinator encourages founders to focus on growth rate rather than absolute numbers. It prevents them from being discouraged early on, when the absolute numbers are still low. It also helps them decide what to focus on: you can use growth rate as a compass to tell you how to evolve the company. But the main advantage is that by focusing on growth rate you tend to get something that grows exponentially.YC doesn't explicitly tell founders that with growth rate "you get out what you put in," but it's not far from the truth. And if growth rate were proportional to performance, then the reward for performance p over time t would be proportional to pt.
+
+Even after decades of thinking about this, I find that sentence startling."""
+
+split_documents = load_and_split_document(text=data)
+root_path = get_storage_root_path("", StorageBackend.GCS)
+vector_store = Chroma(persist_directory=str(root_path / "chromadb"), embedding_function=GPT4AllEmbeddings())
+db = vector_store.add_documents(split_documents)

sandbox_alexis/storage_backend.py

@@ -0,0 +1,15 @@
+from upath import UPath as Path
+from enum import Enum
+
+
+class StorageBackend(Enum):
+    LOCAL = "local"
+    MEMORY = "memory"
+    GCS = "gcs"
+    S3 = "s3"
+    AZURE = "az"
+
+
+def get_storage_root_path(bucket_name, storage_backend: StorageBackend):
+    root_path = Path(f"{storage_backend.value}://{bucket_name}")
+    return root_path

sandbox_alexis/vector_store.py

@@ -0,0 +1,7 @@
+import chromadb
+from storage_backend import StorageBackend, get_storage_root_path
+
+
+root_path = get_storage_root_path("sample_bucket", StorageBackend.GCS)
+client = chromadb.PersistentClient(root_path / "chromadb")
+collection = client.get_or_create_collection("embeddings")