Skip to content

fix: Converting company size to corresponding letter code. #87

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
aniebietafia merged 1 commit into from
Apr 1, 2025
Merged

fix: Converting company size to corresponding letter code. #87

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
55 changes: 16 additions & 39 deletions src/main/java/integrations/telex/salesagent/user/service/OpenAIChatService.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,9 @@ private void handleDetailsInput(String channelId, String message) throws JsonPro
log.info("Lead Details: {}", details);
leadDetailsMap.put(channelId, details);

//details.setCompanySizeClass(classifyCompanySize(details.getCompanySizes()));
String sizeCode = classifyCompanySize(details.getCompanySizes());

log.info("Company size code: {}", sizeCode);

String prompt1 = "Thank you for the details, I'll now conduct research on " +
details.getCompanySizes() + " companies in " + details.getLocations() +
Expand Down Expand Up @@ -134,9 +136,7 @@ private LeadDetails extractLeadDetails(String userInput) throws JsonProcessingEx
""", userInput);

String response = chatModel.call(prompt);
LeadDetails extractedJSON = objectMapper.readValue(response, LeadDetails.class);
log.info("Extracted Object {}", extractedJSON);
return extractedJSON;
return objectMapper.readValue(response, LeadDetails.class);
}

private void generateAndSendResearch(String channelId, LeadDetails details) throws JsonProcessingException {
Expand Down Expand Up @@ -168,36 +168,6 @@ private void generateAndSendResearch(String channelId, LeadDetails details) thro
}
}

private void generateBusinessResearch(String channelId, LeadDetails details) throws JsonProcessingException {
try {
// Generate research
String researchPrompt = String.format("""
Provide a detailed business lead research for %s companies in %s.
Focus on companies that would benefit from %s services.
Include:
1. List of 5-10 potential leads with brief descriptions
2. Key market trends in this sector
3. Recommended outreach approach
""",
details.getCompanySizes(), details.getLocations(), details.getBusinessType());

String research = chatModel.call(researchPrompt);
telexClient.sendInstruction(channelId, research);

// Generate pitch
String pitch = generatePitch(details, channelId);
telexClient.sendInstruction(channelId,
"I've completed the initial research and generated a preliminary list. " +
"Based on the gathered data, I've also drafted a tailored pitch:\n\n" + pitch);

conversationStates.put(channelId, ConversationState.COMPLETE);

} catch (Exception e) {
log.error("Research generation failed", e);
telexClient.sendInstruction(channelId, "I couldn't complete the research. Please try again.");
}
}

private String generatePitch(LeadDetails details, String channelId) throws JsonProcessingException {
exitProcess(channelId);
return String.format("""
Expand Down Expand Up @@ -259,11 +229,18 @@ private boolean isRestartRequest(String message) {
}

private String classifyCompanySize(String companySize) {
if (companySize == null) return "C";
companySize = companySize.toLowerCase();
if (companySize.contains("mid")) return "B";
if (companySize.contains("large")) return "A";
return "C"; // default to small
if (companySize == null || companySize.isEmpty()) {
return "C";
}

companySize = companySize.toLowerCase().trim();

return switch (companySize) {
case "large", "l", "a" -> "A";
case "mid-sized", "mid", "b" -> "B";
case "small", "s", "c" -> "C";
default -> "C";
};
}

private CompanySearchRequest convertToCompanySearchRequest(LeadDetails details) {
Expand Down