From c85bf225b9c1e34b4e347f8fbf880dc9f3244b53 Mon Sep 17 00:00:00 2001
From: Marcus Schiesser <mail@marcusschiesser.de>
Date: Wed, 27 Dec 2023 10:26:15 +0700
Subject: [PATCH] fix: python packaging
---
.../templates/components/vectordbs/python/none/__init__.py | 0
.../templates/types/simple/fastapi/app/api/routers/chat.py | 4 ++--
.../templates/types/simple/fastapi/app/engine/__init__.py | 0
.../templates/types/simple/fastapi/pyproject.toml | 2 +-
.../templates/types/streaming/fastapi/app/engine/__init__.py | 0
.../templates/types/streaming/fastapi/pyproject.toml | 2 +-
6 files changed, 4 insertions(+), 4 deletions(-)
create mode 100644 packages/create-llama/templates/components/vectordbs/python/none/__init__.py
create mode 100644 packages/create-llama/templates/types/simple/fastapi/app/engine/__init__.py
create mode 100644 packages/create-llama/templates/types/streaming/fastapi/app/engine/__init__.py
diff --git a/packages/create-llama/templates/components/vectordbs/python/none/__init__.py b/packages/create-llama/templates/components/vectordbs/python/none/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/packages/create-llama/templates/types/simple/fastapi/app/api/routers/chat.py b/packages/create-llama/templates/types/simple/fastapi/app/api/routers/chat.py
index 9b1b1a1a8..f6a246386 100644
--- a/packages/create-llama/templates/types/simple/fastapi/app/api/routers/chat.py
+++ b/packages/create-llama/templates/types/simple/fastapi/app/api/routers/chat.py
@@ -1,7 +1,7 @@
from typing import List
from fastapi import APIRouter, Depends, HTTPException, status
-from llama_index import VectorStoreIndex
+from llama_index.chat_engine.types import BaseChatEngine
from llama_index.llms.base import ChatMessage
from llama_index.llms.types import MessageRole
from pydantic import BaseModel
@@ -26,7 +26,7 @@ class _Result(BaseModel):
@r.post("")
async def chat(
data: _ChatData,
- chat_engine: VectorStoreIndex = Depends(get_chat_engine),
+ chat_engine: BaseChatEngine = Depends(get_chat_engine),
) -> _Result:
# check preconditions and get last message
if len(data.messages) == 0:
diff --git a/packages/create-llama/templates/types/simple/fastapi/app/engine/__init__.py b/packages/create-llama/templates/types/simple/fastapi/app/engine/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/packages/create-llama/templates/types/simple/fastapi/pyproject.toml b/packages/create-llama/templates/types/simple/fastapi/pyproject.toml
index 10d1d609e..f9bb9605b 100644
--- a/packages/create-llama/templates/types/simple/fastapi/pyproject.toml
+++ b/packages/create-llama/templates/types/simple/fastapi/pyproject.toml
@@ -1,5 +1,5 @@
[tool.poetry]
-name = "llamaindex-fastapi"
+name = "app"
version = "0.1.0"
description = ""
authors = ["Marcus Schiesser <mail@marcusschiesser.de>"]
diff --git a/packages/create-llama/templates/types/streaming/fastapi/app/engine/__init__.py b/packages/create-llama/templates/types/streaming/fastapi/app/engine/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/packages/create-llama/templates/types/streaming/fastapi/pyproject.toml b/packages/create-llama/templates/types/streaming/fastapi/pyproject.toml
index 3a99fabd7..f9bb9605b 100644
--- a/packages/create-llama/templates/types/streaming/fastapi/pyproject.toml
+++ b/packages/create-llama/templates/types/streaming/fastapi/pyproject.toml
@@ -1,5 +1,5 @@
[tool.poetry]
-name = "llamaindex-fastapi-streaming"
+name = "app"
version = "0.1.0"
description = ""
authors = ["Marcus Schiesser <mail@marcusschiesser.de>"]
--
GitLab