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

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", "openai",
"neo4j", "neo4j",
"tiktoken", "tiktoken",
"google-generativeai" "google-generativeai",
], ],
scripts=[ scripts=[
"scripts/chunker-recursive", "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", "response_mime_type": "text/plain",
} }
block_level = HarmBlockThreshold.BLOCK_ONLY_HIGH
self.safety_settings={ self.safety_settings={
HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_ONLY_HIGH, HarmCategory.HARM_CATEGORY_HATE_SPEECH: block_level,
HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_ONLY_HIGH, HarmCategory.HARM_CATEGORY_HARASSMENT: block_level,
HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_ONLY_HIGH, HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: block_level,
HarmCategory.HARM_CATEGORY_DANGEROUS: HarmBlockThreshold.BLOCK_ONLY_HIGH, HarmCategory.HARM_CATEGORY_DANGEROUS: block_level,
HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY: HarmBlockThreshold.BLOCK_ONLY_HIGH, HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY: block_level,
} }
self.llm = genai.GenerativeModel( self.llm = genai.GenerativeModel(