splunk/DECEIVE
1
0
Fork 0
mirror of https://github.com/splunk/DECEIVE.git synced 2025-07-01 16:47:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Removed extraneous debug prints

Browse Source
This commit is contained in:
David J. Bianco
2024-08-22 15:39:49 -04:00
parent 7e38c43dee
commit e385b8a4bb
1 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ssh_server.py
View File

@ -170,21 +170,15 @@ def choose_llm():
model_name = config['llm'].get("model_name", "gpt-3.5-turbo")
if llm_provider_name == 'openai':
print("***** Model: OpenAI")
print("***** Model Name: ", model_name)
llm_model = ChatOpenAI(
model=model_name
)
elif llm_provider_name == 'aws':
print("***** Model: AWS")
print("***** Model Name: ", model_name)
llm_model = ChatBedrockConverse(
model=model_name,
region_name=config['llm'].get("aws_region", "us-east-1"),
credentials_profile_name=config['llm'].get("aws_credentials_profile", "default") )
elif llm_provider_name == 'gemini':
print("***** Model: Gemini")
print("***** Model Name: ", model_name)
llm_model = ChatGoogleGenerativeAI(
model=model_name,
)