fixed llama-index package versions and add missing env for web data s…

…ource (#14)
run-llama · Mar 22, 2024 · 60ed8fe · 60ed8fe
1 parent 56faee0
commit 60ed8fe
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 2 deletions.
diff --git a/.changeset/khaki-jeans-end.md b/.changeset/khaki-jeans-end.md
@@ -0,0 +1,5 @@
+---
+"create-llama": patch
+---
+
+Add missing environment variable config for web data source
diff --git a/.changeset/popular-berries-stare.md b/.changeset/popular-berries-stare.md
@@ -0,0 +1,5 @@
+---
+"create-llama": patch
+---
+
+Fixed llama-index package versions
diff --git a/helpers/env-variables.ts b/helpers/env-variables.ts
@@ -5,6 +5,7 @@ import {
   TemplateDataSource,
   TemplateFramework,
   TemplateVectorDB,
+  WebSourceConfig,
 } from "./types";
 
 type EnvVar = {
@@ -105,18 +106,22 @@ const getDataSourceEnvs = (
 ) => {
   switch (dataSource.type) {
     case "web":
+      const config = dataSource.config as WebSourceConfig;
       return [
         {
           name: "BASE_URL",
           description: "The base URL to start web scraping.",
+          value: config.baseUrl,
         },
         {
           name: "URL_PREFIX",
           description: "The prefix of the URL to start web scraping.",
+          value: config.baseUrl,
         },
         {
           name: "MAX_DEPTH",
           description: "The maximum depth to scrape.",
+          value: config.depth?.toString(),
         },
       ];
     case "file":

diff --git a/templates/types/simple/fastapi/pyproject.toml b/templates/types/simple/fastapi/pyproject.toml
@@ -10,7 +10,9 @@ python = "^3.11,<3.12"
 fastapi = "^0.109.1"
 uvicorn = { extras = ["standard"], version = "^0.23.2" }
 python-dotenv = "^1.0.0"
-llama-index = "^0.10.7"
+llama-index = "0.10.15"
+llama-index-core = "0.10.15"
+llama-index-agent-openai = "0.1.5"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/templates/types/streaming/fastapi/pyproject.toml b/templates/types/streaming/fastapi/pyproject.toml
@@ -10,7 +10,9 @@ python = "^3.11,<3.12"
 fastapi = "^0.109.1"
 uvicorn = { extras = ["standard"], version = "^0.23.2" }
 python-dotenv = "^1.0.0"
-llama-index = "^0.10.7"
+llama-index = "0.10.15"
+llama-index-core = "0.10.15"
+llama-index-agent-openai = "0.1.5"
 
 [build-system]
 requires = ["poetry-core"]