Skip to content

Commit

Permalink
Bumped version of litellm and minor logging
Browse files Browse the repository at this point in the history
  • Loading branch information
vivek-athina committed Dec 11, 2024
1 parent 264f98d commit 8a7ee3d
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 2 deletions.
7 changes: 7 additions & 0 deletions athina/steps/qdrant_retrieval.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,12 +69,19 @@ def execute(self, input_data: Any) -> Union[Dict[str, Any], None]:

try:
response = self._retriever.retrieve(input_text)
if not response:
print("No chunks retrieved")
return {
"status": "success",
"data": [],
}
result = [node.get_content() for node in response]
return {
"status": "success",
"data": result,
}
except Exception as e:
print(f"Error during retrieval: {str(e)}")
return {
"status": "error",
"data": str(e),
Expand Down
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "athina"
version = "1.6.26"
version = "1.6.27"
description = "Python SDK to configure and run evaluations for your LLM-based application"
authors = ["Shiv Sakhuja <shiv@athina.ai>", "Akshat Gupta <akshat@athina.ai>", "Vivek Aditya <vivek@athina.ai>", "Akhil Bisht <akhil@athina.ai>"]
readme = "README.md"
Expand All @@ -21,7 +21,7 @@ python-dotenv = "^1.0.0"
requests = "^2.31.0"
langchain-openai = "^0.0.8"
llama-index = "^0.9.40"
litellm = "1.53.2"
litellm = "1.54.1"
jinja2 = "^3.1.4"
pinecone-client = "^4.1.0"
qdrant-client = "^1.9.1"
Expand Down

0 comments on commit 8a7ee3d

Please sign in to comment.