Feature - Support Azure AI Search as a Vector DB

[maljazaery]

Description

Support Azure AI Search as a vector DB.
Dependencies: azure-search-documents

Type of change

• New feature (non-breaking change which adds functionality)

How Has This Been Tested?

• Test Script (please provide)

from mem0 import Memory
config = {
    "vector_store": {
        "provider": "azure_ai_search",
        "config": {
            "service_name": "xxxx",
            "api_key": "xxx",
            "embedding_model_dims": 1536 ,
        }
    }
}


m = Memory.from_config(config)
m.add("Likes to play cricket on weekends", user_id="alice", metadata={"category": "hobbies"})

Checklist:

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• [] I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes
• [] Any dependent changes have been merged and published in downstream modules
• I have checked my code and corrected any misspellings

Maintainer Checklist

• closes #xxxx (Replace xxxx with the GitHub issue number)
• Made sure Checks passed

[CLAassistant]

All committers have signed the CLA.

[maljazaery]

@Dev-Khant would you take a look please.

[Dev-Khant]

Hey @maljazaery Can you please check why the tests are failing?

[maljazaery]

@Dev-Khant, I fixed the package dependency issue. Now, I am showing an error message if the required python package is not installed. Thanks

[Dev-Khant]

Please run make format to properly format the code. And please can you add tests for this?

[maljazaery]

@Dev-Khant thank you for reviewing the code! I fixed all the issues.

[Dev-Khant]

Hey @maljazaery Thanks for the quick resolution. Can you please also add the tests for this new vectordb?

[maljazaery]

Hi @Dev-Khant, Here is the notebook I used for testing of this feature.

[Dev-Khant]

I checked the notebook and it looks good to me. Thanks @maljazaery for the contribution!

[maljazaery]

[heart] Mohamad Al Jazaery reacted to your message:

…

________________________________ From: Dev Khant ***@***.***> Sent: Tuesday, October 29, 2024 4:42:59 PM To: mem0ai/mem0 ***@***.***> Cc: Mohamad Al Jazaery ***@***.***>; Mention ***@***.***> Subject: Re: [mem0ai/mem0] Feature - Support Azure AI Search as a Vector DB (PR #1967) I checked the notebook and it looks good to me. Thanks @maljazaery<https://github.com/maljazaery> for the contribution! — Reply to this email directly, view it on GitHub<#1967 (comment)>, or unsubscribe<https://github.com/notifications/unsubscribe-auth/BGDTC7M24MYEZSD4I2NHMEDZ563JHAVCNFSM6AAAAABQCP4FKCVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDINBUHAYTCOJWG4>. You are receiving this because you were mentioned.Message ID: ***@***.***>