dummy_agent.py

from typing import List, Dict

class DummyResponseAgent(object):
    def __init__(self):
        """ Load your model(s) here """
        pass

    def generate_responses(self, test_data: List[Dict]) -> List[str]:
        """
        You will be provided with a batch of upto 50 independent conversations
        Return a string for every conversation
        
        Input 1
        [
            {"persona A": ..., "persona B": ... "dialogue": ... }, # conversation 1  Turn 1
            ...
            {"persona A": ..., "persona B": ... "dialogue": ... }  # conversation 50 Turn 1
        ]

        Model should return 50 responses for Turn 1

        ...
        Input 7
        [
            {"persona A": ..., "persona B": ... "dialogue": ... }, # conversation 1  Turn 7
            ...
            {"persona A": ..., "persona B": ... "dialogue": ... }  # conversation 50 Turn 7
        ]
        Model should return 50 responses for Turn 7

        Note: Turn numbers will NOT be provided as input

        """
        # print(f"{len(test_data)=}, {test_data[0].keys()=}, {len(test_data[-1]['dialogue'])}")
        return ["THIS IS A TEST REPLY" for _ in test_data]