Added variable for max
tokens
This commit is contained in:
		
							
								
								
									
										3
									
								
								bot.py
									
									
									
									
									
								
							
							
						
						
									
										3
									
								
								bot.py
									
									
									
									
									
								
							| @@ -8,6 +8,7 @@ from dotenv import load_dotenv | ||||
| load_dotenv() | ||||
| DISCORD_TOKEN = os.getenv('DISCORD_TOKEN') | ||||
| OPENAI_API_KEY = os.getenv('OPENAI_API_KEY') | ||||
| MAX_TOKENS = int(os.getenv('MAX_TOKENS')) | ||||
| aclient = AsyncOpenAI(api_key=OPENAI_API_KEY) | ||||
| # Initialize OpenAI API | ||||
|  | ||||
| @@ -34,7 +35,7 @@ async def on_message(message): | ||||
| async def ask_gpt(ctx, *, question: str): | ||||
|     try: | ||||
|         # Send the question to GPT-4o using the new OpenAI API | ||||
|         response = await aclient.chat.completions.create(model="gpt-4o", max_tokens=2000,  | ||||
|         response = await aclient.chat.completions.create(model="gpt-4o", max_tokens=MAX_TOKENS,  | ||||
|         messages=[ | ||||
|             {"role": "system", "content": "You are a helpful assistant. You exist inside a discord server. There is a character limit so please keep responses brief if possible."}, | ||||
|             {"role": "user", "content": question} | ||||
|   | ||||
		Reference in New Issue
	
	Block a user