| # config.py | |
| # System message to set the assistant's behavior | |
| SYSTEM_MESSAGE = "You are a helpful assistant." | |
| # Model name (update if you have access to Gemini model) | |
| MODEL_NAME = 'gemini-pro' # Use 'gemini-pro' if you have access | |
| MODEL_VISION_NAME = 'gemini-pro-vision' | |
| # Default parameters for AI model generation | |
| DEFAULT_MAX_NEW_TOKENS = 512 | |
| DEFAULT_TEMPERATURE = 0.7 | |
| DEFAULT_TOP_P = 0.95 | |
| DEFAULT_TOP_K = 40 | |
| DEFAULT_STOP_SEQUENCES = None # Can be a list like ['STOP', 'END'] | |
| # Other configuration settings can be added here as needed | |
| # config.py | |
| # System message to set the assistant's behavior | |
| SYSTEM_INSTRUCTION = "You are a helpful assistant." | |
| # Model name (update if you have access to Gemini model) | |
| MODEL_NAME = 'gemini-1.5-flash' # Use the appropriate model name | |
| # Default parameters for AI model generation | |
| DEFAULT_MAX_OUTPUT_TOKENS = 512 | |
| DEFAULT_TEMPERATURE = 0.7 | |
| DEFAULT_TOP_P = 0.95 | |
| DEFAULT_TOP_K = 40 | |
| DEFAULT_STOP_SEQUENCES = None # Can be a list like ['STOP', 'END'] | |
| # Other configuration settings can be added here as needed | |