apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-17 01:01:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fixed missed comma

Browse source
This commit is contained in:
JackColquitt 2024-10-08 11:14:05 -07:00 • committed by Cyber MacGeddon
parent b98e553965
commit 6bfa01d580
2 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-flow/setup.py
View file

@ -55,7 +55,7 @@ setuptools.setup(
"openai",
"neo4j",
"tiktoken",
"google-generativeai"
"google-generativeai",
],
scripts=[
"scripts/chunker-recursive",

12
trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py
View file

@ -74,12 +74,14 @@ class Processor(ConsumerProducer):
"response_mime_type": "text/plain",
}
block_level = HarmBlockThreshold.BLOCK_ONLY_HIGH
self.safety_settings={
HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_ONLY_HIGH,
HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
HarmCategory.HARM_CATEGORY_DANGEROUS: HarmBlockThreshold.BLOCK_ONLY_HIGH,
HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY: HarmBlockThreshold.BLOCK_ONLY_HIGH,
HarmCategory.HARM_CATEGORY_HATE_SPEECH: block_level,
HarmCategory.HARM_CATEGORY_HARASSMENT: block_level,
HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: block_level,
HarmCategory.HARM_CATEGORY_DANGEROUS: block_level,
HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY: block_level,
}
self.llm = genai.GenerativeModel(