def get_gemini_model_endpoint()

in cloud-run-gemini-chat/Application/CloudRun/Python-Flask/gcp_gemini_utils.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def get_gemini_model_endpoint(model):
	''' Determine which Gemini model to use and return the REST URL '''

	host = "https://generativelanguage.googleapis.com"

	#---------------------------------------------------------------------
	# Validate model
	#---------------------------------------------------------------------

	if model is None or len(model) == 0:
		# Default to model (Gemini 1.0 Pro)
		model = "gemini_1_0_pro_latest"

	#---------------------------------------------------------------------
	# Select the Gemini LLM model to use
	#---------------------------------------------------------------------

	if model == "gemini_1_0_pro_latest":
		path = "/v1beta/models/gemini-1.0-pro-latest:generateContent"
	elif model == "gemini_1_0_ultra_latest":
		path = "/v1beta/models/gemini-1.0-ultra-latest:generateContent"
	elif model == "gemini_1_5_pro_latest":
		path = "/v1beta/models/gemini-1.5-pro-latest:generateContent"
	else:
		# Default model (Gemini 1.0 Pro)
		path = "/v1beta/models/gemini-pro:generateContent"

	return host + path