my-project-public

repository

loading code, commits, and activity

repositories

loading repo index

#1	from typing import Any, Callable, List, Optional
#2
#3	from mem0.configs.llms.base import BaseLlmConfig
#4
#5
#6	class OpenAIConfig(BaseLlmConfig):
#7	"""
#8	Configuration class for OpenAI and OpenRouter-specific parameters.
#9	Inherits from BaseLlmConfig and adds OpenAI-specific settings.
#10	"""
#11
#12	def __init__(
#13	self,
#14	# Base parameters
#15	model: Optional[str] = None,
#16	temperature: float = 0.1,
#17	api_key: Optional[str] = None,
#18	max_tokens: int = 2000,
#19	top_p: float = 0.1,
#20	top_k: int = 1,
#21	enable_vision: bool = False,
#22	vision_details: Optional[str] = "auto",
#23	http_client_proxies: Optional[dict] = None,
#24	# OpenAI-specific parameters
#25	openai_base_url: Optional[str] = None,
#26	models: Optional[List[str]] = None,
#27	route: Optional[str] = "fallback",
#28	openrouter_base_url: Optional[str] = None,
#29	site_url: Optional[str] = None,
#30	app_name: Optional[str] = None,
#31	store: bool = False,
#32	# Response monitoring callback
#33	response_callback: Optional[Callable[[Any, dict, dict], None]] = None,
#34	):
#35	"""
#36	Initialize OpenAI configuration.
#37
#38	Args:
#39	model: OpenAI model to use, defaults to None
#40	temperature: Controls randomness, defaults to 0.1
#41	api_key: OpenAI API key, defaults to None
#42	max_tokens: Maximum tokens to generate, defaults to 2000
#43	top_p: Nucleus sampling parameter, defaults to 0.1
#44	top_k: Top-k sampling parameter, defaults to 1
#45	enable_vision: Enable vision capabilities, defaults to False
#46	vision_details: Vision detail level, defaults to "auto"
#47	http_client_proxies: HTTP client proxy settings, defaults to None
#48	openai_base_url: OpenAI API base URL, defaults to None
#49	models: List of models for OpenRouter, defaults to None
#50	route: OpenRouter route strategy, defaults to "fallback"
#51	openrouter_base_url: OpenRouter base URL, defaults to None
#52	site_url: Site URL for OpenRouter, defaults to None
#53	app_name: Application name for OpenRouter, defaults to None
#54	response_callback: Optional callback for monitoring LLM responses.
#55	"""
#56	# Initialize base parameters
#57	super().__init__(
#58	model=model,
#59	temperature=temperature,
#60	api_key=api_key,
#61	max_tokens=max_tokens,
#62	top_p=top_p,
#63	top_k=top_k,
#64	enable_vision=enable_vision,
#65	vision_details=vision_details,
#66	http_client_proxies=http_client_proxies,
#67	)
#68
#69	# OpenAI-specific parameters
#70	self.openai_base_url = openai_base_url
#71	self.models = models
#72	self.route = route
#73	self.openrouter_base_url = openrouter_base_url
#74	self.site_url = site_url
#75	self.app_name = app_name
#76	self.store = store
#77
#78	# Response monitoring
#79	self.response_callback = response_callback
#80

z6Mkq5mY3JWtxoxUobWcfNHm7AkRubgSWEZTkBVqZXJviFZ5/my-project-public