admin/Akern-Genai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

medium thinking level

Browse Source
This commit is contained in:
Matteo Rosati
2026-01-29 10:23:23 +01:00
parent 7c2318c655
commit 111e47cb77
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
llm_config.py
View File

@@ -9,7 +9,8 @@ tools = [
types.Tool( types.Tool(
retrieval=types.Retrieval( retrieval=types.Retrieval(
vertex_rag_store=types.VertexRagStore( vertex_rag_store=types.VertexRagStore(
rag_resources=[types.VertexRagStoreRagResource(rag_corpus=CORPUS)], rag_resources=[
types.VertexRagStoreRagResource(rag_corpus=CORPUS)],
) )
) )
) )
@@ -20,17 +21,19 @@ generate_content_config = types.GenerateContentConfig(
top_p=0.95, top_p=0.95,
max_output_tokens=65535, max_output_tokens=65535,
safety_settings=[ safety_settings=[
types.SafetySetting(category="HARM_CATEGORY_HATE_SPEECH", threshold="OFF"), types.SafetySetting(
category="HARM_CATEGORY_HATE_SPEECH", threshold="OFF"),
types.SafetySetting( types.SafetySetting(
category="HARM_CATEGORY_DANGEROUS_CONTENT", threshold="OFF" category="HARM_CATEGORY_DANGEROUS_CONTENT", threshold="OFF"
), ),
types.SafetySetting( types.SafetySetting(
category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold="OFF" category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold="OFF"
), ),
types.SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold="OFF"), types.SafetySetting(
category="HARM_CATEGORY_HARASSMENT", threshold="OFF"),
], ],
tools=tools, tools=tools,
thinking_config=types.ThinkingConfig( thinking_config=types.ThinkingConfig(
thinking_level="HIGH", thinking_level="MEDIUM",
), ),
) )