added goal achievment preditions

This commit is contained in:
2024-09-17 22:39:07 +00:00
parent 47a274741f
commit 1bfc773782
10 changed files with 325 additions and 20 deletions
+21 -2
View File
@@ -40,7 +40,7 @@ response = bot.predict_based_on_past_assessment(
companyid="testid"
)
print(f"Result: {response}")'''
print(f"Result: {response}")
from src.services.sop_document_parser import DocumentParser
@@ -59,4 +59,23 @@ res = parser.extract_sops_for_workers_by_department(
depts_workers=workers_department
)
print(res)
print(res)'''
from src.services.chatbot import Chatbot
bot = Chatbot(
)
company_info = {
"company_name": "Example Co",
"company_size": "Medium",
"departments": ["HR", "Finance", "IT"]
}
response = bot.predict_goal_achievement_probability(
company_info=company_info,
companyid="testid"
)
print(response)