Forráskód Böngészése

[Bug fix] Fix issue related to get_data_sources() method (#1295)

Deshraj Yadav 1 éve
szülő
commit
91044ec591
2 módosított fájl, 11 hozzáadás és 7 törlés
  1. 10 6
      embedchain/app.py
  2. 1 1
      pyproject.toml

+ 10 - 6
embedchain/app.py

@@ -10,9 +10,14 @@ import requests
 import yaml
 from tqdm import tqdm
 
-from embedchain.cache import (Config, ExactMatchEvaluation,
-                              SearchDistanceEvaluation, cache,
-                              gptcache_data_manager, gptcache_pre_function)
+from embedchain.cache import (
+    Config,
+    ExactMatchEvaluation,
+    SearchDistanceEvaluation,
+    cache,
+    gptcache_data_manager,
+    gptcache_pre_function,
+)
 from embedchain.client import Client
 from embedchain.config import AppConfig, CacheConfig, ChunkerConfig
 from embedchain.core.db.database import get_session, init_db, setup_engine
@@ -21,8 +26,7 @@ from embedchain.embedchain import EmbedChain
 from embedchain.embedder.base import BaseEmbedder
 from embedchain.embedder.openai import OpenAIEmbedder
 from embedchain.evaluation.base import BaseMetric
-from embedchain.evaluation.metrics import (AnswerRelevance, ContextRelevance,
-                                           Groundedness)
+from embedchain.evaluation.metrics import AnswerRelevance, ContextRelevance, Groundedness
 from embedchain.factory import EmbedderFactory, LlmFactory, VectorDBFactory
 from embedchain.helpers.json_serializable import register_deserializable
 from embedchain.llm.base import BaseLlm
@@ -295,7 +299,7 @@ class App(EmbedChain):
         data_sources = self.db_session.query(DataSource).filter_by(app_id=self.local_id).all()
         results = []
         for row in data_sources:
-            results.append({"data_type": row.data_type, "data_value": row.data_value, "metadata": row.metadata})
+            results.append({"data_type": row.type, "data_value": row.value, "metadata": row.meta_data})
         return results
 
     def deploy(self):

+ 1 - 1
pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "embedchain"
-version = "0.1.89"
+version = "0.1.90"
 description = "Simplest open source retrieval(RAG) framework"
 authors = [
     "Taranjeet Singh <taranjeet@embedchain.ai>",