[version] bump package version and minor cleanup (#909)

This commit is contained in:
Deshraj Yadav
2023-11-05 16:34:36 -08:00
committed by GitHub
parent 830a7397ef
commit f0d112254b
2 changed files with 1 additions and 32 deletions

View File

@@ -7,7 +7,6 @@ import uuid
import requests import requests
import yaml import yaml
from fastapi import FastAPI, HTTPException
from embedchain import Client from embedchain import Client
from embedchain.config import PipelineConfig from embedchain.config import PipelineConfig
@@ -396,33 +395,3 @@ class Pipeline(EmbedChain):
yaml_path=yaml_path, yaml_path=yaml_path,
auto_deploy=auto_deploy, auto_deploy=auto_deploy,
) )
def start(self, host="0.0.0.0", port=8000):
app = FastAPI()
@app.post("/add")
async def add_document(data_value: str, data_type: str = None):
"""
Add a document to the pipeline.
"""
try:
document = {"data_value": data_value, "data_type": data_type}
self.add(document)
return {"message": "Document added successfully"}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.post("/query")
async def query_documents(query: str, num_documents: int = 3):
"""
Query for similar documents in the pipeline.
"""
try:
results = self.search(query, num_documents)
return results
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
import uvicorn
uvicorn.run(app, host=host, port=port)

View File

@@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "embedchain" name = "embedchain"
version = "0.0.91" version = "0.0.92"
description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data" description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data"
authors = [ authors = [
"Taranjeet Singh <taranjeet@embedchain.ai>", "Taranjeet Singh <taranjeet@embedchain.ai>",