def results_from_questions()

in backend-apis/app/utils/utils_salesforce.py [0:0]

• 34 lines of code
• 8 McCabe index (conditional complexity)

def results_from_questions(questions: list, conversation_id: str) -> list:
    """Search for results from questions.

    Args:
        questions:
            Questions to be searched

    Returns:
        list
            List of dict representations of results
    """
    results = []
    for question in questions:
        if (
            "image" in question.lower()
            or "picture" in question.lower()
            or "attach" in question.lower()
        ):
            continue

        try:
            search_results = utils_search.vertexai_search_multiturn(
                search_query=question,
                conversation_id=conversation_id,
                summary_result_count=2,
                datastore_id=config["website_search"]["website_datastore_id"],
            )
            search_results_dict = Message.to_dict(search_results)
        except GoogleAPICallError as e:
            print(e)
            continue

        result = {}
        result["question"] = question
        result["summary_text"] = search_results_dict.get("reply", {}).get(
            "reply", ""
        )
        result["links"] = []

        for i, r in enumerate(search_results_dict.get("search_results", [])):
            if i == 2:
                break
            result["links"].append(
                config["salesforce"]["website_uri"] + r["id"]
            )
        results.append(result)

    return results