18 lines
928 B
Python
18 lines
928 B
Python
from django.db import models
|
|
|
|
class Neo4jProfile(models.Model):
|
|
"""
|
|
Stores multiple Neo4j database configurations for different use cases, along with AI model configurations.
|
|
"""
|
|
name = models.CharField(max_length=100, unique=True, help_text="Profile name for the Neo4j database.")
|
|
uri = models.CharField(max_length=255, help_text="Neo4j connection URI.")
|
|
username = models.CharField(max_length=100, help_text="Neo4j username.")
|
|
password = models.CharField(max_length=100, help_text="Neo4j password (stored securely).")
|
|
|
|
# AI Model Configuration for Cypher Query Generation
|
|
openai_api_key = models.CharField(max_length=255, help_text="OpenAI API key (or equivalent LLM provider).")
|
|
model_name = models.CharField(max_length=100, help_text="LLM Model used (e.g., gpt-4, deepseek-chat).")
|
|
|
|
def __str__(self):
|
|
return f"{self.name} | Model: {self.model_name}"
|