Files
semantic-kernel/python/semantic_kernel/__init__.py
T
9dbbf56c96 Python: Add astra memory (#4374)
### Motivation and Context

We are going to add
[Astra](https://docs.datastax.com/en/home/docs/index.html) as a memory
store.

DataStax Astra DB is a serverless vector database that’s perfect for
managing mission-critical AI workloads. It’s built on Apache Cassandra®,
making Astra DB a highly scalable, reliable database technology. It’s a
powerful all-in-one data storage solution, ideal for Generative AI
projects.

### Description

- Add astra into python/semantic_kernel/connectors/memory/astradb
- Add test into
python/tests/integration/connectors/memory/test_astradb.py

### Contribution Checklist

<!-- Before submitting this PR, please make sure: -->

- [ ] The code builds clean without any errors or warnings
- [ ] The PR follows the [SK Contribution
Guidelines](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md)
and the [pre-submission formatting
script](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md#development-scripts)
raises no violations
- [ ] All unit tests pass, and I have added new tests where possible
- [ ] I didn't break anyone 😄

---------

Co-authored-by: jsbilgi <[email protected]>
Co-authored-by: Obioma Anomnachi <[email protected]>
2024-01-30 21:51:04 +00:00

59 lines
2.1 KiB
Python

# Copyright (c) Microsoft. All rights reserved.
from semantic_kernel import core_plugins, memory
from semantic_kernel.kernel import Kernel
from semantic_kernel.orchestration.context_variables import ContextVariables
from semantic_kernel.orchestration.kernel_context import KernelContext
from semantic_kernel.orchestration.kernel_function_base import KernelFunctionBase
from semantic_kernel.semantic_functions.chat_prompt_template import ChatPromptTemplate
from semantic_kernel.semantic_functions.prompt_template import PromptTemplate
from semantic_kernel.semantic_functions.prompt_template_config import (
PromptTemplateConfig,
)
from semantic_kernel.semantic_functions.semantic_function_config import (
SemanticFunctionConfig,
)
from semantic_kernel.utils.logging import setup_logging
from semantic_kernel.utils.null_logger import NullLogger
from semantic_kernel.utils.settings import (
astradb_settings_from_dot_env,
azure_aisearch_settings_from_dot_env,
azure_aisearch_settings_from_dot_env_as_dict,
azure_cosmos_db_settings_from_dot_env,
azure_openai_settings_from_dot_env,
bing_search_settings_from_dot_env,
google_palm_settings_from_dot_env,
mongodb_atlas_settings_from_dot_env,
openai_settings_from_dot_env,
pinecone_settings_from_dot_env,
postgres_settings_from_dot_env,
redis_settings_from_dot_env,
)
__all__ = [
"Kernel",
"NullLogger",
"azure_cosmos_db_settings_from_dot_env",
"openai_settings_from_dot_env",
"azure_openai_settings_from_dot_env",
"azure_aisearch_settings_from_dot_env",
"azure_aisearch_settings_from_dot_env_as_dict",
"postgres_settings_from_dot_env",
"pinecone_settings_from_dot_env",
"astradb_settings_from_dot_env",
"bing_search_settings_from_dot_env",
"mongodb_atlas_settings_from_dot_env",
"google_palm_settings_from_dot_env",
"redis_settings_from_dot_env",
"PromptTemplateConfig",
"PromptTemplate",
"ChatPromptTemplate",
"SemanticFunctionConfig",
"ContextVariables",
"KernelFunctionBase",
"KernelContext",
"memory",
"core_plugins",
"setup_logging",
]