chrissewell/DECEIVE
1
0
Fork 0
mirror of https://github.com/ChrisSewell/DECEIVE.git synced 2025-07-01 18: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") model_name = config['llm'].get("model_name", "gpt-3.5-turbo")
if llm_provider_name == 'openai': if llm_provider_name == 'openai':
print("***** Model: OpenAI")
print("***** Model Name: ", model_name)
llm_model = ChatOpenAI( llm_model = ChatOpenAI(
model=model_name model=model_name
) )
elif llm_provider_name == 'aws': elif llm_provider_name == 'aws':
print("***** Model: AWS")
print("***** Model Name: ", model_name)
llm_model = ChatBedrockConverse( llm_model = ChatBedrockConverse(
model=model_name, model=model_name,
region_name=config['llm'].get("aws_region", "us-east-1"), region_name=config['llm'].get("aws_region", "us-east-1"),
credentials_profile_name=config['llm'].get("aws_credentials_profile", "default") ) credentials_profile_name=config['llm'].get("aws_credentials_profile", "default") )
elif llm_provider_name == 'gemini': elif llm_provider_name == 'gemini':
print("***** Model: Gemini")
print("***** Model Name: ", model_name)
llm_model = ChatGoogleGenerativeAI( llm_model = ChatGoogleGenerativeAI(
model=model_name, model=model_name,
) )